Update app.py

app.py

@@ -95,13 +95,16 @@ 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, download_path = loop.run_until_complete(infer(model_str, prompt, nprompt, height, width, steps, cfg, seed, inference_timeout))
+        result = 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, download_path
+    return result
 
 
 def add_gallery(image, model_str, gallery):
@@ -137,7 +140,7 @@ with gr.Blocks(fill_width=True, head=js) as demo:
         with gr.Column(scale=1):
             with gr.Group():
                 with gr.Row():
-                    output = [gr.Image(label=m, show_download_button=True, elem_classes="output",
+                    output = [gr.Image(label=m, show_download_button=True, elem_classes=["renderedImg"],
                               interactive=False, width=112, height=112, show_share_button=False, format="png",
                               visible=True) for m in default_models]
                     current_models = [gr.Textbox(m, visible=False) for m in default_models]