Update app.py

app.py

@@ -45,7 +45,7 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
         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
-     except:
+    except:
         try:
             print('n_1')
             with closing(create_connection(f"{url_sd3}", timeout=60)) as conn:
@@ -61,20 +61,20 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
                 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
+        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():