Translation any to english

app.py

@@ -6,7 +6,12 @@ from diffusers import AutoPipelineForText2Image, AutoPipelineForImage2Image
 #from diffusers.utils import load_image
 import torch
 from PIL import Image
+from huggingface_hub import hf_hub_download
+from eSeNTranslate import TranslateFromAny2XModel
 
+# Load translation model(s) (Pipeline)
+fasttextModelPath = hf_hub_download(repo_id="facebook/fasttext-language-identification", filename="model.bin")
+translatePipe = TranslateFromAny2XModel(nllb_model_path="facebook/nllb-200-distilled-600M", fasttext_model_path=fasttextModelPath)
 
 modelPath = "stabilityai/sdxl-turbo"
  
@@ -40,6 +45,7 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
         seed = random.randint(0, MAX_SEED)
         
     generator = torch.Generator().manual_seed(seed)
+    prompt = translatePipe.generate(prompt)
     
     if use_as_input:
         print("Image to Image:")