Update app.py

app.py

@@ -39,14 +39,18 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
     
     url_sd5 = os.getenv("url_sd5")
     url_sd6 = os.getenv("url_sd6")
-    
-    data = {"inputs":f"{prompt}, 4k photo","options":{"negative_prompt":"[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry","width":1024,"height":1024,"guidance_scale":7,"num_inference_steps":35}}
-    response = requests.post(f'{url_sd5}', json=data)
-    print(response.text)
-    print(response.json()['image']['file_name'])
-    file_name = response.json()['image']['file_name']
-    photo = f"{url_sd6}{file_name}.png"
-    return photo
+    try:
+        data = {"inputs":f"{prompt}, 4k photo","options":{"negative_prompt":"[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry","width":1024,"height":1024,"guidance_scale":7,"num_inference_steps":35}}
+        response = requests.post(f'{url_sd5}', json=data)
+        print(response.text)
+        print(response.json()['image']['file_name'])
+        file_name = response.json()['image']['file_name']
+        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_sd3}", timeout=60)) as conn:
 #    conn.send('{"fn_index":3,"session_hash":""}')