Spaces:
Sleeping
Sleeping
rollback to googletrans
Browse files
app.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
|
| 2 |
-
|
| 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 |
-
|
| 130 |
-
|
| 131 |
-
|
| 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 |
|