Update app.py

app.py

@@ -61,15 +61,16 @@ async def infer(model_index, prompt, nprompt="", height=0, width=0, steps=0, cfg
     return None
 
 def gen_fn(model_index, prompt, nprompt="", height=0, width=0, steps=0, cfg=0, seed=-1):
-    try:
+        try:
         loop = asyncio.new_event_loop()
-        result = loop.run_until_complete(infer(model_index, prompt, nprompt,
-                                         height, width, steps, cfg, seed, inference_timeout))
+        result = loop.run_until_complete(infer(model_str, prompt, seed, inference_timeout))
     except (Exception, asyncio.CancelledError) as e:
         print(e)
-        print(f"Task aborted: {models[model_index]}")
+        print(f"Task aborted: {model_str}")
         result = None
-        raise gr.Error(f"Task aborted: {models[model_index]}, Error: {e}")
+        with lock:
+            image = "https://huggingface.co/spaces/Yntec/HuggingfaceDiffusion/resolve/main/protest.png"
+        result = image
     finally:
         loop.close()
     return result