Upload 2 files

app.py

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