Prueba_09

app.py

@@ -1,10 +1,13 @@
-import sentencepiece
 import gradio as gr
-from transformers import pipeline
+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")
 
 def translate(name):
 
-    trans = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es", tokenizer="Helsinki-NLP/opus-mt-en-es")
+    trans = pipeline("translation", model, tokenizer)
     return trans(name)[0]["translation_text"]
 
 demo = gr.Interface(fn=translate, inputs="text", outputs="text")

requirements.txt

@@ -1,4 +1,3 @@
 transformers
 gradio
-torch
-sentencepiece
+torch