Update app.py

app.py

@@ -32,23 +32,23 @@ default_models = models[:num_models]
 inference_timeout = 600
 
 #async def infer(model_str, prompt, seed=1, timeout=inference_timeout):
-    kwargs = {"seed": seed}
-    task = asyncio.create_task(asyncio.to_thread(models_load[model_str].fn, prompt=prompt, **kwargs, token=HF_TOKEN))
-    await asyncio.sleep(0)
-    try:
-        result = await asyncio.wait_for(task, timeout=timeout)
-    except (Exception, asyncio.TimeoutError) as e:
-        print(e)
-        print(f"Task timed out: {model_str}")
-        if not task.done(): 
-            task.cancel()
-        result = None
-    if task.done() and result is not None:
-        with lock:
-            png_path = "image.png"
-            result.save(png_path)
-        return png_path
-    return None
+  #  kwargs = {"seed": seed}
+    #task = asyncio.create_task(asyncio.to_thread(models_load[model_str].fn, prompt=prompt, **kwargs, token=HF_TOKEN))
+ #   await asyncio.sleep(0)
+ #   try:
+    #    result = await asyncio.wait_for(task, timeout=timeout)
+  #  except (Exception, asyncio.TimeoutError) as e:
+   #     print(e)
+    #    print(f"Task timed out: {model_str}")
+     #   if not task.done(): 
+     #       task.cancel()
+     #   result = None
+ #   if task.done() and result is not None:
+     #   with lock:
+     #       png_path = "image.png"
+       #     result.save(png_path)
+  #      return png_path
+    #return None