juancamval commited on
Commit
7a20655
·
verified ·
1 Parent(s): 27ede32

Update app.py

Browse files

Eliminacion de tts

Files changed (1) hide show
  1. app.py +9 -23
app.py CHANGED
@@ -1,28 +1,20 @@
1
  from transformers import pipeline
2
  import gradio as gr
3
- import os # Importa la librería 'os' para acceder a las variables de entorno
4
 
5
- # Modelo de traducción inglés-español (generalmente público)
6
  modelo_traduccion = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es")
7
 
8
- # Modelo de Text-to-Speech para español (generalmente público)
9
- modelo_tts = pipeline("text-to-speech", model="espnet/spanish_tts_tacotron2")
10
-
11
  def traducir_texto(texto_en):
12
  """
13
- Traduce directamente el texto de entrada del inglés al español y genera audio.
14
- Lee el token de Hugging Face del entorno (si está configurado).
15
  """
16
- api_token = os.environ.get("HF_TOKEN") # Lee el valor de la variable de entorno HF_TOKEN
17
  if not texto_en.strip():
18
- return "", None
19
  try:
20
- resultado_traduccion = modelo_traduccion(texto_en)[0]['translation_text']
21
- # Generar audio a partir del texto traducido
22
- audio_output = modelo_tts(resultado_traduccion)
23
- return resultado_traduccion, audio_output['audio']
24
  except Exception as e:
25
- return f"Error en la traducción o TTS: {e}", None
26
 
27
  def borrar_entrada(texto_entrada):
28
  """
@@ -32,24 +24,18 @@ def borrar_entrada(texto_entrada):
32
 
33
  with gr.Blocks() as interfaz_traductor:
34
  """
35
- Define la interfaz gráfica de la aplicación de traducción de inglés a español con Text-to-Speech.
36
- Lee el token de Hugging Face del entorno (si está configurado).
37
  """
38
- gr.Markdown("# Traductor Inglés a Español con TTS")
39
  with gr.Row():
40
  with gr.Column():
41
  texto_entrada_en = gr.Textbox(label="Texto en inglés", placeholder="Pega aquí el texto en inglés", lines=10)
42
  boton_borrar = gr.Button("Borrar todo")
43
  with gr.Column():
44
  texto_salida_es = gr.Textbox(label="Traducción al español", placeholder="La traducción aparecerá aquí", interactive=False, lines=10)
45
- audio_salida = gr.Audio(label="Audio de la Traducción")
46
  gr.Markdown("*Puedes copiar el texto traducido seleccionándolo y usando Ctrl+C (o Cmd+C en macOS).*")
47
 
48
- def actualizar_salida(texto_entrada):
49
- traduccion, audio = traducir_texto(texto_entrada)
50
- return traduccion, audio
51
-
52
- texto_entrada_en.change(actualizar_salida, inputs=texto_entrada_en, outputs=[texto_salida_es, audio_salida])
53
  boton_borrar.click(borrar_entrada, inputs=texto_entrada_en, outputs=texto_entrada_en)
54
 
55
  interfaz_traductor.launch()
 
1
  from transformers import pipeline
2
  import gradio as gr
 
3
 
4
+ # Define el modelo de traducción inglés-español
5
  modelo_traduccion = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es")
6
 
 
 
 
7
  def traducir_texto(texto_en):
8
  """
9
+ Traduce directamente el texto de entrada del inglés al español.
 
10
  """
 
11
  if not texto_en.strip():
12
+ return ""
13
  try:
14
+ resultado = modelo_traduccion(texto_en)
15
+ return resultado[0]['translation_text']
 
 
16
  except Exception as e:
17
+ return f"Error en la traducción: {e}"
18
 
19
  def borrar_entrada(texto_entrada):
20
  """
 
24
 
25
  with gr.Blocks() as interfaz_traductor:
26
  """
27
+ Define la interfaz gráfica de la aplicación de traducción de inglés a español (básica).
 
28
  """
29
+ gr.Markdown("# Traductor Inglés a Español")
30
  with gr.Row():
31
  with gr.Column():
32
  texto_entrada_en = gr.Textbox(label="Texto en inglés", placeholder="Pega aquí el texto en inglés", lines=10)
33
  boton_borrar = gr.Button("Borrar todo")
34
  with gr.Column():
35
  texto_salida_es = gr.Textbox(label="Traducción al español", placeholder="La traducción aparecerá aquí", interactive=False, lines=10)
 
36
  gr.Markdown("*Puedes copiar el texto traducido seleccionándolo y usando Ctrl+C (o Cmd+C en macOS).*")
37
 
38
+ texto_entrada_en.change(traducir_texto, inputs=texto_entrada_en, outputs=texto_salida_es)
 
 
 
 
39
  boton_borrar.click(borrar_entrada, inputs=texto_entrada_en, outputs=texto_entrada_en)
40
 
41
  interfaz_traductor.launch()