Prueba_10

app.py

@@ -1,13 +1,11 @@
+import sentencepiece
 import gradio as gr
-from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
-
-tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-es")
-
-model = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-es")
+from transformers import pipeline
 
+print (dir(sentencepiece))
 def translate(name):
 
-    trans = pipeline("translation", model, tokenizer)
+    trans = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es", tokenizer="Helsinki-NLP/opus-mt-en-es")
     return trans(name)[0]["translation_text"]
 
 demo = gr.Interface(fn=translate, inputs="text", outputs="text")