juancamval commited on
Commit
d5693d8
verified
1 Parent(s): be5de5f

Cargar el modelo directamente

Browse files

Cambio de pipeline a AutoTokenizer, AutoModelForSeq2SeqLM

Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,8 +1,8 @@
1
- from transformers import pipeline
2
  import gradio as gr
3
 
4
- # Define el modelo de traducci贸n ingl茅s-espa帽ol
5
- modelo_traduccion = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es")
6
 
7
  def traducir_texto(texto_en):
8
  """
 
1
+ from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
2
  import gradio as gr
3
 
4
+ tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-es")
5
+ model = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-en-es")
6
 
7
  def traducir_texto(texto_en):
8
  """