Update app.py

app.py

@@ -112,24 +112,43 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
                 result = client.predict(prompt,negative_prompt,"","",True,False,False,0,1024,1024,7,1,25,25,False,api_name="/run")
                 return result
             except:
-                ua = UserAgent()
-                headers = {
-                    'authority': 'radames-real-time-text-to-image-sdxl-lightning.hf.space',
-                    'accept': 'text/event-stream',
-                    'accept-language': 'ru,en;q=0.9,la;q=0.8,ja;q=0.7',
-                    'cache-control': 'no-cache',
-                    'referer': 'https://radames-real-time-text-to-image-sdxl-lightning.hf.space/?__theme=light',
-                    'sec-ch-ua': '"Not_A Brand";v="8", "Chromium";v="120", "YaBrowser";v="24.1", "Yowser";v="2.5"',
-                    'sec-ch-ua-mobile': '?0',
-                    'sec-ch-ua-platform': '"Windows"',
-                    'sec-fetch-dest': 'empty',
-                    'sec-fetch-mode': 'cors',
-                    'sec-fetch-site': 'same-origin',
-                    'user-agent': f'{ua.random}'
-                }
-                client = Client("radames/Real-Time-Text-to-Image-SDXL-Lightning", headers=headers)
-                result = client.predict(prompt, [], 0, random.randint(1, 999999), fn_index=0)
-                return result
+                try:
+                    ua = UserAgent()
+                    headers = {
+                        'authority': 'radames-real-time-text-to-image-sdxl-lightning.hf.space',
+                        'accept': 'text/event-stream',
+                        'accept-language': 'ru,en;q=0.9,la;q=0.8,ja;q=0.7',
+                        'cache-control': 'no-cache',
+                        'referer': 'https://radames-real-time-text-to-image-sdxl-lightning.hf.space/?__theme=light',
+                        'sec-ch-ua': '"Not_A Brand";v="8", "Chromium";v="120", "YaBrowser";v="24.1", "Yowser";v="2.5"',
+                        'sec-ch-ua-mobile': '?0',
+                        'sec-ch-ua-platform': '"Windows"',
+                        'sec-fetch-dest': 'empty',
+                        'sec-fetch-mode': 'cors',
+                        'sec-fetch-site': 'same-origin',
+                        'user-agent': f'{ua.random}'
+                    }
+                    client = Client("radames/Real-Time-Text-to-Image-SDXL-Lightning", headers=headers)
+                    result = client.predict(prompt, [], 0, random.randint(1, 999999), fn_index=0)
+                    return result
+                except:
+                    try:
+                        ua = UserAgent()
+                        headers = {
+                            'user-agent': f'{ua.random}'
+                        }
+                        client = Client("https://ashrafb-arpr.hf.space/", headers=headers)
+                        result = client.predict(prompt, fn_index=0)
+                        return result
+                    except:
+                        ua = UserAgent()
+                        headers = {
+                            'user-agent': f'{ua.random}'
+                        }
+                        client = Client("https://ashrafb-arv3s.hf.space/", headers=headers)
+                        result = client.predict(prompt,0,"Cinematic", fn_index=0)
+                        return result
+                        
 
 
 def mirror(image_output, scale_by, method, gfpgan, codeformer):