Update app.py

app.py

@@ -24,11 +24,11 @@ else:
     pipe = DiffusionPipeline.from_pretrained(model_id, use_safetensors=True)
     pipe = pipe.to(device)
 
-def he_to_en(prompt):
-    html_str = requests.get( url = "https://translate.google.com", params = {"sl": "iw", "tl": "en", "op": "translate", "text": prompt} ).text()
+def translate(text,lang):
+    html_str = requests.get( url = "https://translate.google.com", params = {"sl": "auto", "tl": lang, "op": "translate", "text": text} ).text()
     tree = fromstring(html_str)
-    english = tree.xpath('span[lang="en"]/span/span/text()')[0]
-    return english
+    translated = tree.xpath('span[lang="'+lang+'"]/span/span/text()')[0]
+    return translated
 
 def generate_random_string(length):
     characters = string.ascii_letters + string.digits
@@ -37,7 +37,7 @@ def generate_random_string(length):
 @spaces.GPU(25)
 def infer(prompt):
     name = generate_random_string(12)+".png"
-    image = pipe(he_to_en(prompt)).images[0].save(name)
+    image = pipe(translate(prompt,"en")).images[0].save(name)
     return name
 
 css="""