Update app.py

app.py

@@ -11,7 +11,7 @@ from PIL import Image
 import io
 import base64
 import re
-
+from gradio_client import Client
 
 def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
     result = {"prompt": prompt,"negative_prompt": negative_prompt,"task": task,"steps": steps,"sampler": sampler,"cfg_scale": cfg_scale,"seed": seed}
@@ -40,34 +40,39 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
     print(task)
     
     try:
-        print('n_1')
-        with closing(create_connection(f"{url_sd3}", timeout=60)) 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":""}}')
-            while True:
-                status = json.loads(conn.recv())['msg']
-                if status == 'estimation':
-                    continue
-                if status == 'process_starts':
-                    break
-            photo = json.loads(conn.recv())['output']['data'][0][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
+        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
     except:
-        print("n_2")
-        print(url_sd4)
-        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
+        try:
+            print('n_1')
+            with closing(create_connection(f"{url_sd3}", timeout=60)) 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":""}}')
+                while True:
+                    status = json.loads(conn.recv())['msg']
+                    if status == 'estimation':
+                        continue
+                    if status == 'process_starts':
+                        break
+                photo = json.loads(conn.recv())['output']['data'][0][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
+        except:
+            print("n_2")
+            print(url_sd4)
+            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
 
 
 def flipp():