M2M100 translator

app.py

@@ -1,6 +1,7 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 #from googletrans import Translator
-from deep_translator import GoogleTranslator
+#from deep_translator import GoogleTranslator
+from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
 import datetime
 import requests
 import pytz
@@ -125,8 +126,23 @@ def native_language_translator(text: str, target_lang: str) -> str:
         target_lang: Target language code (e.g., 'be', 'es', 'fr', 'zh')
     """
     try:
-        translation = GoogleTranslator(source='auto', target=target_lang).translate(text)
-        return f"Translated to {target_lang}: {translation}"
+        # Load the model and tokenizer
+        model_name = "facebook/m2m100_418M"
+        tokenizer = M2M100Tokenizer.from_pretrained(model_name)
+        model = M2M100ForConditionalGeneration.from_pretrained(model_name)
+
+        # Set the target language
+        tokenizer.src_lang = "en"  # Assume input is in English
+        tokenizer.tgt_lang = target_lang
+
+        # Tokenize the input text
+        encoded_text = tokenizer(text, return_tensors="pt")
+
+        # Generate translation
+        generated_tokens = model.generate(
+            **encoded_text,
+            forced_bos_token_id=tokenizer.get_lang_id(target_lang)
+        translation = tokenizer.decode(generated_tokens[0], skip_special_tokens=True)
     except Exception as e:
         return f"Translation failed: {str(e)}"