Upload app.py

app.py

@@ -61,11 +61,10 @@ async def infer(model_index, prompt, nprompt="", height=None, width=None, steps=
         if not task.done(): task.cancel()
         result = None
         raise Exception(e)
-    if task.done() and result is not None:
+    if task.done() and result is not None and not isinstance(result, tuple):
         with lock:
             png_path = "image.png"
-            if isinstance(result, tuple): result[0].save(png_path)
-            else: result.save(png_path)
+            result.save(png_path)
             image = str(Path(png_path).resolve())
         return image
     return None