Update app.py

app.py

@@ -48,9 +48,17 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
         photo = f"{url_sd6}{file_name}.png"
         return photo
     except:
-        client = Client("https://prodia-sdxl-stable-diffusion-xl.hf.space")
-        result = client.predict(prompt,"[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry","sd_xl_base_1.0.safetensors [be9edd61]",25,"DPM++ 2M Karras",7,1024,1024,-1,fn_index=0)
-        return result
+       with closing(create_connection(f"{url_sd4}", timeout=60)) 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()
+            conn.recv()
+            conn.recv()
+            conn.recv()
+            photo = json.loads(conn.recv())['output']['data'][0]
+            photo = photo.replace('data:image/jpeg;base64,', '').replace('data:image/png;base64,', '')
+            photo = Image.open(io.BytesIO(base64.decodebytes(bytes(photo, "utf-8"))))
+            return photo
     
 #with closing(create_connection(f"{url_sd3}", timeout=60)) as conn:
 #    conn.send('{"fn_index":3,"session_hash":""}')