Update app.py

app.py

@@ -95,7 +95,8 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0
         theSeed = seed
     kwargs["seed"] = theSeed
         
-    task = asyncio.create_task(asyncio.to_thread(models_load[model_str].fn, prompt=prompt, negative_prompt=nprompt, **kwargs, token=HF_TOKEN))
+    # task = asyncio.create_task(asyncio.to_thread(models_load[model_str].fn, prompt=prompt, negative_prompt=nprompt, **kwargs, token=HF_TOKEN))
+    task = asyncio.create_task(asyncio.to_thread(models_load[model_str].fn, prompt=prompt, negative_prompt=nprompt, **kwargs))
     await asyncio.sleep(0)
     try:
         result = await asyncio.wait_for(task, timeout=timeout)
@@ -121,8 +122,7 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0
 def gen_fn(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0, seed=-1):
     try:
         loop = asyncio.new_event_loop()
-        result = loop.run_until_complete(infer(model_str, prompt, nprompt,
-                                         height, width, steps, cfg, seed, inference_timeout))
+        result = loop.run_until_complete(infer(model_str, prompt, nprompt, height, width, steps, cfg, seed, inference_timeout))
     except (Exception, asyncio.CancelledError) as e:
         print(e)
         print(f"gen_fn: Task aborted: {model_str}")