Update app.py

app.py

@@ -28,7 +28,7 @@ async def infer(model_str, prompt, seed=-1):
         asyncio.to_thread(models_load[model_str].fn, prompt=prompt, seed=seed, token=HF_TOKEN)
     )
     await asyncio.sleep(0)
-    result = await asyncio.wait_for(task, timeout=300)
+    result = await asyncio.wait_for(task, timeout=600)
     return result