Update app.py

app.py

@@ -61,16 +61,17 @@ 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_str, prompt, seed, inference_timeout))
+        result = loop.run_until_complete(infer(model_index, prompt, nprompt,
+                                         height, width, steps, cfg, seed, inference_timeout))
     except (Exception, asyncio.CancelledError) as e:
         print(e)
-        print(f"Task aborted: {model_str}")
-        result = None
+        print(f"Task aborted: {models[model_index]}")with lock:
         with lock:
-            image = "https://huggingface.co/spaces/Yntec/HuggingfaceDiffusion/resolve/main/protest.png"
+            image = "https://huggingface.co/spaces/Yntec/ToyWorld/resolve/main/error.png"
         result = image
+        raise gr.Error(f"Task aborted: {models[model_index]}, Error: {e}")
     finally:
         loop.close()
     return result
@@ -96,7 +97,7 @@ with gr.Blocks(theme='John6666/YntecDark', fill_width=True, css=css) as myface:
         <div class="center"><h1>Blitz Diffusion</h1></div>
         <p style="margin-bottom: 1px; color: #ffaa66;">
         <h3>{int(len(models))} Stable Diffusion models, but why? For your enjoyment!</h3></p>
-        <br><div class="wrapper">11.21 <img src="https://huggingface.co/Yntec/DucHaitenLofi/resolve/main/NEW.webp" alt="NEW!" style="width:32px;height:16px;">This has become a legacy backup copy of old <u><a href="https://huggingface.co/spaces/Yntec/ToyWorld">ToyWorld</a></u>'s UI! Newer models added dailty over there! 8 new models since last update!</div>
+        <br><div class="wrapper">12.21 <img src="https://huggingface.co/Yntec/DucHaitenLofi/resolve/main/NEW.webp" alt="NEW!" style="width:32px;height:16px;">This has become a legacy backup copy of old <u><a href="https://huggingface.co/spaces/Yntec/ToyWorld">ToyWorld</a></u>'s UI! Newer models added dailty over there! 19 new models since last update!</div>
         <p style="margin-bottom: 1px; font-size: 98%">
         <br><h4>If a model is already loaded each new image takes less than <b>10</b> seconds to generate!</h4></p>
         <p style="margin-bottom: 1px; color: #ffffff;">