Update app.py
Browse files
app.py
CHANGED
|
@@ -1,11 +1,11 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
-
from transformers import
|
| 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 =
|
| 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
|
| 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 |
]
|