Update app.py

app.py

@@ -66,16 +66,19 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0
     if steps > 0: kwargs["num_inference_steps"] = steps
     if cfg > 0: cfg = kwargs["guidance_scale"] = cfg
 
+    print(f"infer: model '{model_str}', prompt: '{prompt}'...")
     if seed == -1:
         theSeed = randomize_seed()
     else: 
         theSeed = seed
     kwargs["seed"] = theSeed
         
+    print("Create task...")
     task = asyncio.create_task(asyncio.to_thread(models_load[model_str].fn, prompt=prompt, negative_prompt=nprompt, **kwargs, token=HF_TOKEN))
     await asyncio.sleep(0)
     try:
         result = await asyncio.wait_for(task, timeout=timeout)
+        print(f"Task ended, result: {result}")
     except asyncio.TimeoutError as e:
         print(e)
         print(f"infer: Task timed out: {model_str}")
@@ -93,6 +96,8 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0
             png_path =  model_str.replace("/", "_") + " - " + get_current_time() + "_" + str(theSeed) + ".png"
             image = save_image(result, png_path, model_str, prompt, nprompt, height, width, steps, cfg, theSeed)
         return image
+    else:
+        print(f"Else, no valid result...: result '{result}'")
     return None
 
 def gen_fn(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0, seed=-1):