Update app.py

app.py

@@ -75,12 +75,13 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0
         result = await asyncio.wait_for(task, timeout=timeout)
     except asyncio.TimeoutError as e:
         print(e)
-        print(f"Task timed out: {model_str}")
+        print(f"infer: Task timed out: {model_str}")
         if not task.done(): task.cancel()
         result = None
         raise Exception(f"Task timed out: {model_str}") from e
     except Exception as e:
         print(e)
+        print(f"infer: exception: {model_str}")
         if not task.done(): task.cancel()
         result = None
         raise Exception() from e
@@ -98,7 +99,7 @@ def gen_fn(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0, see
                                          height, width, steps, cfg, seed, inference_timeout))
     except (Exception, asyncio.CancelledError) as e:
         print(e)
-        print(f"Task aborted: {model_str}")
+        print(f"gen_fn: Task aborted: {model_str}")
         result = None
         raise gr.Error(f"Task aborted: {model_str}, Error: {e}")
     finally: