Bryam182 commited on
Commit
06a1c83
1 Parent(s): ae89794

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -9
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
- boton_limpiar.click(
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()