Update app.py

app.py

@@ -7,7 +7,7 @@ from diffusers import StableDiffusionPipeline as DiffusionPipeline
 import torch
 from pathos.multiprocessing import ProcessingPool as ProcessPoolExecutor
 import requests
-from lxml.html.soupparser import fromstring
+from lxml import etree
 
 pool = ProcessPoolExecutor(4)
 pool.__enter__()
@@ -15,16 +15,6 @@ 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)
@@ -34,13 +24,22 @@ else:
     pipe = pipe.to(device)
 
 def translate(text,lang):
+    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'
+    ]
     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]
+    tree = etree.HTML(html_str)
+    translated = tree.xpath(f'//span[@lang="{lang}"]/span/span[text()]')
     return translated
 
 def generate_random_string(length):
@@ -49,7 +48,9 @@ def generate_random_string(length):
 
 def infer(prompt):
     name = generate_random_string(12)+".png"
-    image = pipe(translate(prompt,"en")).images[0].save(name)
+    english_prompt = translate(prompt,"en")
+    print(f'Final prompt: {english_prompt}')
+    image = pipe(english_prompt).images[0].save(name)
     return name
 
 css="""