Update app.py
Browse files
app.py
CHANGED
@@ -74,7 +74,7 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, cfg=0, seed=-1
|
|
74 |
if task.done() and result is not None and not isinstance(result, tuple):
|
75 |
with lock:
|
76 |
png_path = "image.png"
|
77 |
-
image = save_image(result, png_path, model_str, prompt, nprompt, height, width,
|
78 |
return image
|
79 |
return None
|
80 |
|
|
|
74 |
if task.done() and result is not None and not isinstance(result, tuple):
|
75 |
with lock:
|
76 |
png_path = "image.png"
|
77 |
+
image = save_image(result, png_path, model_str, prompt, nprompt, height, width, cfg, seed)
|
78 |
return image
|
79 |
return None
|
80 |
|