Update app.py

app.py

@@ -80,7 +80,7 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
             return photo
         
     try:
-        with closing(create_connection(f"{url_sd3}")) as conn:
+        with closing(create_connection(f"{url_sd3}", timeout=30)) as conn:
             conn.send('{"fn_index":3,"session_hash":""}')
             conn.send(f'{{"data":["{prompt}, 4k photo","[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry",7.5,"(No style)"],"event_data":null,"fn_index":3,"session_hash":""}}')
             c = 0
@@ -97,7 +97,7 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
             photo = Image.open(io.BytesIO(base64.decodebytes(bytes(photo, "utf-8"))))
             return photo
     except:
-        with closing(create_connection(f"{url_sd4}")) as conn:
+        with closing(create_connection(f"{url_sd4}", timeout=30)) as conn:
             conn.send('{"fn_index":0,"session_hash":""}')
             conn.send(f'{{"data":["{prompt}","[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry","dreamshaperXL10_alpha2.safetensors [c8afe2ef]",30,"DPM++ 2M Karras",7,1024,1024,-1],"event_data":null,"fn_index":0,"session_hash":""}}')
             conn.recv()