revamped code of translate.py

app.py

@@ -52,7 +52,7 @@ interface = gr.Interface(
         gr.Image(label="Generated Image"),
     ],
     title="LinguaVision - Multilingual Chatbot with Image Generation",
-    description="Enter text in any language to translate it into English and generate an image based on the text.\n NOTE: This tool takes approximately 20 minutes to execute.",
+    description="Enter text in any language to translate it into English and generate an image based on the text.",
 )
 
 if __name__ == "__main__":

translate.py

@@ -40,9 +40,9 @@ def translate_text(text, src_lang="auto", tgt_lang="en"):
     # Tokenize the input text
     inputs = tokenizer(text, return_tensors="pt", padding=True)
     
-    # Set the source language and target language for the model
-    # If source language is auto, the model will automatically detect it
-    tokenizer.src_lang = src_lang if src_lang != "auto" else None
+    # Set the source and target languages
+    if src_lang != "auto":
+        tokenizer.src_lang = src_lang
     tokenizer.tgt_lang = tgt_lang
 
     # Generate the translation