prol99 commited on
Commit
39d0228
·
verified ·
1 Parent(s): 60e80b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,11 +1,11 @@
1
  import gradio as gr
2
- from transformers import AutoModelForSeq2SeqGeneration, AutoTokenizer
3
  import torch
4
 
5
  # Cargar el modelo y tokenizador directamente en lugar de usar pipeline
6
  model_name = "projecte-aina/aina-translator-es-ast"
7
  tokenizer = AutoTokenizer.from_pretrained(model_name)
8
- model = AutoModelForSeq2SeqGeneration.from_pretrained(model_name)
9
 
10
  def translate_text(text):
11
  """
@@ -41,7 +41,7 @@ demo = gr.Interface(
41
  title="Traductor Español-Asturiano",
42
  description="Traductor basado en el modelo AINA para traducir del español al asturiano.",
43
  examples=[
44
- ["Hola, ¿cómo vvvestás?"],
45
  ["Me gusta mucho Asturias y su cultura"],
46
  ["El cielo está muy azul hoy"]
47
  ]
 
1
  import gradio as gr
2
+ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
3
  import torch
4
 
5
  # Cargar el modelo y tokenizador directamente en lugar de usar pipeline
6
  model_name = "projecte-aina/aina-translator-es-ast"
7
  tokenizer = AutoTokenizer.from_pretrained(model_name)
8
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
9
 
10
  def translate_text(text):
11
  """
 
41
  title="Traductor Español-Asturiano",
42
  description="Traductor basado en el modelo AINA para traducir del español al asturiano.",
43
  examples=[
44
+ ["Hola, ¿cómo jjjestás?"],
45
  ["Me gusta mucho Asturias y su cultura"],
46
  ["El cielo está muy azul hoy"]
47
  ]