Update app.py

app.py

@@ -40,41 +40,34 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
     print(task)
     
     try:
-        data = f'{"inputs": {prompt},"style":"(No style)"}'
-        response = requests.post('https://enzostvs-stable-diffusion-tpu.hf.space/api', data=data)
-        file_name = response.json()['image']['file_name']
-        photo = f"https://huggingface.co/datasets/enzostvs/stable-diffusion-tpu-generations/resolve/main/images/{file_name}.png"
-        return photo
+        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:
-        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
+        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():