feat: add lang support

app.py

@@ -7,7 +7,8 @@ from PIL import Image
 import re
 import os
 
-
+from libretranslatepy import LibreTranslateAPI
+lt = LibreTranslateAPI("https://translate.argosopentech.com/")
 
 auth_token = os.getenv("auth_token")
 model_id = "CompVis/stable-diffusion-v1-4"
@@ -15,10 +16,14 @@ device = "cpu"
 pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=auth_token)
 pipe = pipe.to(device)
 
-def infer(prompt, samples, steps, scale, seed):     
+def infer(prompt, samples, steps, scale, seed):       
+    lang_detected = lt.detect(prompt)["language]"
+    print(lang_detected)
+    translated = lt.translate(prompt, lang_detected, "en")
+    print(translated)
     generator = torch.Generator(device=device).manual_seed(seed)
     images_list = pipe(
-        [prompt] * samples,
+        [translated] * samples,
         num_inference_steps=steps,
         guidance_scale=scale,
         generator=generator,

requirements.txt

@@ -10,4 +10,5 @@ scipy
 scikit-image
 diffusers==0.3.0
 transformers
-ftfy
+ftfy
+libretranslatepy