Update app.py

app.py

@@ -95,16 +95,13 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0
 def gen_fn(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0, seed=-1):
     try:
         loop = asyncio.new_event_loop()
-        result = loop.run_until_complete(infer(model_str, prompt, nprompt,
-                                         height, width, steps, cfg, seed, inference_timeout))
+        result, download_path = loop.run_until_complete(infer(model_str, prompt, nprompt, height, width, steps, cfg, seed, inference_timeout))
     except (Exception, asyncio.CancelledError) as e:
-        print(e)
         print(f"gen_fn: Task aborted: {model_str}")
-        result = None
         raise gr.Error(f"Task aborted: {model_str}, Error: {e}")
     finally:
         loop.close()
-    return result
+    return result, download_path
 
 
 def add_gallery(image, model_str, gallery):