Update app.py

app.py

@@ -13,9 +13,18 @@ pool = ProcessPoolExecutor(4)
 pool.__enter__()
 
 model_id = "runwayml/stable-diffusion-v1-5"
-
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
+user_agents = [
+    'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/109.0.0.0 Safari/537.36'
+    'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/109.0.0.0 Safari/537.36'
+    'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36'
+    'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36'
+    'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36'
+    'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.1 Safari/605.1.15'
+    'Mozilla/5.0 (Macintosh; Intel Mac OS X 13_1) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.1 Safari/605.1.15'
+]
+
 if torch.cuda.is_available():
     torch.cuda.max_memory_allocated(device=device)
     pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
@@ -25,9 +34,13 @@ else:
     pipe = pipe.to(device)
 
 def translate(text,lang):
-    html_str = requests.get( url = "http://translate.google.com", params = {"sl": "auto", "tl": lang, "op": "translate", "text": text}, headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/111.0.0.0 Safari/537.36'} ).text
+    html_str = requests.get(
+        url = "http://translate.google.com",
+        params = {"sl": "auto", "tl": lang, "op": "translate", "text": text},
+        headers = {'User-Agent': random.choice(user_agents)}
+    ).text
     tree = fromstring(html_str)
-    translated = tree.xpath('//span[@lang="'+lang+'"]/span/span/text()')[0]
+    translated = tree.xpath('//span[@lang="'+lang+'"]/span/span[text()]')[0]
     return translated
 
 def generate_random_string(length):