Update app.py
Browse files
app.py
CHANGED
@@ -5,20 +5,16 @@ from transformers import pipeline
|
|
5 |
traductor = pipeline(task='text2text-generation', model='facebook/m2m100_418M')
|
6 |
|
7 |
def traducir_texto(texto, idioma_destino):
|
8 |
-
"""Traduce el texto al idioma especificado."""
|
9 |
respuesta = traductor(texto, forced_bos_token_id=traductor.tokenizer.get_lang_id(lang=idioma_destino))
|
10 |
return respuesta[0]["generated_text"]
|
11 |
|
12 |
def traducir_ingles_a_espanol(texto):
|
13 |
-
"""Traduce el texto del inglés al español."""
|
14 |
return traducir_texto(texto, "es")
|
15 |
|
16 |
def traducir_espanol_a_ingles(texto):
|
17 |
-
"""Traduce el texto del español al inglés."""
|
18 |
return traducir_texto(texto, "en")
|
19 |
|
20 |
def limpiar_entrada(texto):
|
21 |
-
"""Limpia el texto de entrada."""
|
22 |
return ""
|
23 |
|
24 |
with gr.Blocks() as demo:
|
@@ -34,11 +30,7 @@ with gr.Blocks() as demo:
|
|
34 |
with gr.Row():
|
35 |
boton_espanol_ingles = gr.Button("Español a Inglés")
|
36 |
|
37 |
-
|
38 |
-
fn=limpiar_entrada,
|
39 |
-
inputs=entrada_texto,
|
40 |
-
outputs=entrada_texto)
|
41 |
-
|
42 |
boton_ingles_espanol.click(
|
43 |
fn=traducir_ingles_a_espanol,
|
44 |
inputs=entrada_texto,
|
@@ -48,5 +40,11 @@ with gr.Blocks() as demo:
|
|
48 |
fn=traducir_espanol_a_ingles,
|
49 |
inputs=entrada_texto,
|
50 |
outputs=etiqueta_respuesta)
|
|
|
|
|
|
|
|
|
|
|
|
|
51 |
|
52 |
demo.launch()
|
|
|
5 |
traductor = pipeline(task='text2text-generation', model='facebook/m2m100_418M')
|
6 |
|
7 |
def traducir_texto(texto, idioma_destino):
|
|
|
8 |
respuesta = traductor(texto, forced_bos_token_id=traductor.tokenizer.get_lang_id(lang=idioma_destino))
|
9 |
return respuesta[0]["generated_text"]
|
10 |
|
11 |
def traducir_ingles_a_espanol(texto):
|
|
|
12 |
return traducir_texto(texto, "es")
|
13 |
|
14 |
def traducir_espanol_a_ingles(texto):
|
|
|
15 |
return traducir_texto(texto, "en")
|
16 |
|
17 |
def limpiar_entrada(texto):
|
|
|
18 |
return ""
|
19 |
|
20 |
with gr.Blocks() as demo:
|
|
|
30 |
with gr.Row():
|
31 |
boton_espanol_ingles = gr.Button("Español a Inglés")
|
32 |
|
33 |
+
|
|
|
|
|
|
|
|
|
34 |
boton_ingles_espanol.click(
|
35 |
fn=traducir_ingles_a_espanol,
|
36 |
inputs=entrada_texto,
|
|
|
40 |
fn=traducir_espanol_a_ingles,
|
41 |
inputs=entrada_texto,
|
42 |
outputs=etiqueta_respuesta)
|
43 |
+
|
44 |
+
boton_limpiar.click(
|
45 |
+
fn=limpiar_entrada,
|
46 |
+
inputs=entrada_texto,
|
47 |
+
outputs=entrada_texto)
|
48 |
+
|
49 |
|
50 |
demo.launch()
|