Update app.py

app.py

@@ -46,7 +46,7 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
     hf_token = os.getenv("hf_token")
     if task == "Playground v2":
         playground = str(os.getenv("playground"))
-        with closing(create_connection("wss://ashrafb-arpr.hf.space/queue/join")) as conn:
+        with closing(create_connection("wss://ashrafb-arpr.hf.space/queue/join", timeout=60)) as conn:
             conn.send('{"fn_index":0,"session_hash":""}')
             conn.send(f'{{"fn_index":0,"data":["{prompt}"],"session_hash":""}}')
             conn.recv()
@@ -62,7 +62,7 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
             
     if task == "Artigen v3":
         artigen = str(os.getenv("artigen"))
-        with closing(create_connection("wss://ashrafb-arv3s.hf.space/queue/join")) as conn:
+        with closing(create_connection("wss://ashrafb-arv3s.hf.space/queue/join", timeout=60)) as conn:
             conn.send('{"fn_index":0,"session_hash":""}')
             conn.send(f'{{"fn_index":0,"data":["{prompt}", 0, "No style"],"session_hash":""}}')
             conn.recv()