Fix FlaskUI initialization

This commit is contained in:
Daya Adianto 2022-12-02 11:04:35 +07:00 committed by Lincoln Stein
parent 6b5d19347a
commit a78a8728fe

View File

@ -207,9 +207,10 @@ class InvokeAIWebServer:
FlaskUI( FlaskUI(
app=self.app, app=self.app,
socketio=self.socketio, socketio=self.socketio,
server="flask_socketio", start_server="flask-socketio",
width=1600, width=1600,
height=1000, height=1000,
idle_interval=10,
port=self.port port=self.port
).run() ).run()
except KeyboardInterrupt: except KeyboardInterrupt:
@ -250,7 +251,7 @@ class InvokeAIWebServer:
return candidate return candidate
assert "Frontend files cannot be found. Cannot continue" assert "Frontend files cannot be found. Cannot continue"
def setup_app(self): def setup_app(self):
self.result_url = "outputs/" self.result_url = "outputs/"
self.init_image_url = "outputs/init-images/" self.init_image_url = "outputs/init-images/"
@ -775,10 +776,10 @@ class InvokeAIWebServer:
).convert("RGBA") ).convert("RGBA")
""" """
The outpaint image and mask are pre-cropped by the UI, so the bounding box we pass The outpaint image and mask are pre-cropped by the UI, so the bounding box we pass
to the generator should be: to the generator should be:
{ {
"x": 0, "x": 0,
"y": 0, "y": 0,
"width": original_bounding_box["width"], "width": original_bounding_box["width"],
"height": original_bounding_box["height"] "height": original_bounding_box["height"]
@ -798,7 +799,7 @@ class InvokeAIWebServer:
) )
""" """
Apply the mask to the init image, creating a "mask" image with Apply the mask to the init image, creating a "mask" image with
transparency where inpainting should occur. This is the kind of transparency where inpainting should occur. This is the kind of
mask that prompt2image() needs. mask that prompt2image() needs.
""" """