File size: 756 Bytes
b50c516 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | import gradio as gr
from transformers import pipeline
translator = pipeline(
"translation",
model="projecte-aina/aina-translator-es-ast",
# Parámetros para una traducción más literal
do_sample=False, # Desactiva la generación aleatoria
num_beams=1, # Usa búsqueda greedy
temperature=0.7, # Temperatura más baja para traducciones más conservadoras
max_length=200 # Limita la longitud para evitar generación extra
)
def translate(text):
if not text:
return ""
result = translator(text)
return result[0]['translation_text']
demo = gr.Interface(
fn=translate,
inputs="text",
outputs="text",
title="Traductor Español-Asturiano"
)
if __name__ == "__main__":
demo.launch() |