Fix example 5

This commit is contained in:
Moishe Lettvin
2024-01-17 18:58:03 -05:00
parent 13f2f792af
commit 0d21768d00
4 changed files with 19 additions and 28 deletions

View File

@@ -13,7 +13,6 @@ class HuggingFaceAIService(AIService):
# available models at https://huggingface.co/Helsinki-NLP (**not all models use 2-character language codes**)
def run_text_translation(self, sentence, source_language, target_language):
translator = pipeline(f"translation", model=f"Helsinki-NLP/opus-mt-{source_language}-{target_language}")
print(translator(sentence))
return translator(sentence)[0]["translation_text"]