KoRiF commited on
Commit
035d806
·
verified ·
1 Parent(s): 9d519fd

rollback to googletrans

Browse files
Files changed (1) hide show
  1. app.py +4 -20
app.py CHANGED
@@ -1,5 +1,5 @@
1
  from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
2
- #from googletrans import Translator
3
  #from deep_translator import GoogleTranslator
4
  from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
5
  import datetime
@@ -126,25 +126,9 @@ def native_language_translator(text: str, target_lang: str) -> str:
126
  target_lang: Target language code (e.g., 'be', 'es', 'fr', 'zh')
127
  """
128
  try:
129
- # Load the model and tokenizer
130
- model_name = "facebook/m2m100_418M"
131
- tokenizer = M2M100Tokenizer.from_pretrained(model_name)
132
- model = M2M100ForConditionalGeneration.from_pretrained(model_name)
133
-
134
- # Set the target language
135
- tokenizer.src_lang = "en" # Assume input is in English
136
- tokenizer.tgt_lang = target_lang
137
-
138
- # Tokenize the input text
139
- encoded_text = tokenizer(text, return_tensors="pt")
140
-
141
- # Generate translation
142
- generated_tokens = model.generate(
143
- **encoded_text,
144
- forced_bos_token_id=tokenizer.get_lang_id(target_lang)
145
- )
146
- translation = tokenizer.decode(generated_tokens[0], skip_special_tokens=True)
147
-
148
  except Exception as e:
149
  return f"Translation failed: {str(e)}"
150
 
 
1
  from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
2
+ from googletrans import Translator
3
  #from deep_translator import GoogleTranslator
4
  from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
5
  import datetime
 
126
  target_lang: Target language code (e.g., 'be', 'es', 'fr', 'zh')
127
  """
128
  try:
129
+ translator = Translator()
130
+ translation = translator.translate(text, dest=target_lang)
131
+ return f"Translated to {target_lang}: {translation.text}"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
132
  except Exception as e:
133
  return f"Translation failed: {str(e)}"
134