Update app.py

app.py

@@ -13,6 +13,7 @@ import base64
 import re
 from gradio_client import Client
 from random import randrange
+from fake_useragent import UserAgent
 
 
 def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
@@ -40,6 +41,7 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
             return photo
     except:
         try:
+            ua = UserAgent()
             headers = {
                 'authority': 'ehristoforu-stable-cascade.hf.space',
                 'accept': 'text/event-stream',
@@ -52,12 +54,13 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
                 'sec-fetch-dest': 'empty',
                 'sec-fetch-mode': 'cors',
                 'sec-fetch-site': 'same-origin',
-                'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 YaBrowser/24.1.0.0 Safari/537.36',
+                'user-agent': f'{ua.random}'
             }
             client = Client("https://ehristoforu-stable-cascade.hf.space", headers=headers)
             result = client.predict(prompt, negative_prompt, 1024, 1024, True)
             return result[0]['image']
         except:
+            ua = UserAgent()
             headers = {
                 'authority': 'multimodalart-stable-cascade.hf.space',
                 'accept': 'text/event-stream',
@@ -70,7 +73,7 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
                 'sec-fetch-dest': 'empty',
                 'sec-fetch-mode': 'cors',
                 'sec-fetch-site': 'same-origin',
-                'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 YaBrowser/24.1.0.0 Safari/537.36',
+                'user-agent':  f'{ua.random}'
             }
             client = Client("multimodalart/stable-cascade", headers=headers)
             result = client.predict(prompt, negative_prompt, randrange(100000), 1024,	1024, 20, 4, 10, 0, 1, api_name="/run")