gelezeta commited on
Commit
37f0e92
·
verified ·
1 Parent(s): 3a0d6a7

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -1,6 +1,7 @@
1
  from transformers import MarianMTModel, MarianTokenizer
2
  import gradio as gr
3
 
 
4
  modelos = {
5
  "Español a Inglés": "Helsinki-NLP/opus-mt-es-en",
6
  "Español a Francés": "Helsinki-NLP/opus-mt-es-fr",
@@ -9,25 +10,28 @@ modelos = {
9
  "Español a Portugués": "Helsinki-NLP/opus-mt-es-pt"
10
  }
11
 
 
12
  def traducir(texto, idioma_destino):
13
  modelo = modelos[idioma_destino]
14
  tokenizer = MarianTokenizer.from_pretrained(modelo)
15
  model = MarianMTModel.from_pretrained(modelo)
16
-
17
  tokens = tokenizer(texto, return_tensors="pt", padding=True)
18
  traduccion = model.generate(**tokens)
19
  resultado = tokenizer.decode(traduccion[0], skip_special_tokens=True)
20
  return resultado
21
 
 
22
  interfaz = gr.Interface(
23
  fn=traducir,
24
  inputs=[
25
  gr.Textbox(lines=5, label="Texto en Español"),
26
- gr.Dropdown(choices=list(modelos.keys()), label="Traducir a:", value="Español a Inglés")
27
  ],
28
  outputs="text",
29
  title="🌍 Traductor Multilingüe",
30
- description="Selecciona un idioma destino y traduce desde español"
31
  )
32
 
 
33
  interfaz.launch()
 
1
  from transformers import MarianMTModel, MarianTokenizer
2
  import gradio as gr
3
 
4
+ # Diccionario con modelos de traducción
5
  modelos = {
6
  "Español a Inglés": "Helsinki-NLP/opus-mt-es-en",
7
  "Español a Francés": "Helsinki-NLP/opus-mt-es-fr",
 
10
  "Español a Portugués": "Helsinki-NLP/opus-mt-es-pt"
11
  }
12
 
13
+ # Función de traducción
14
  def traducir(texto, idioma_destino):
15
  modelo = modelos[idioma_destino]
16
  tokenizer = MarianTokenizer.from_pretrained(modelo)
17
  model = MarianMTModel.from_pretrained(modelo)
18
+
19
  tokens = tokenizer(texto, return_tensors="pt", padding=True)
20
  traduccion = model.generate(**tokens)
21
  resultado = tokenizer.decode(traduccion[0], skip_special_tokens=True)
22
  return resultado
23
 
24
+ # Interfaz con Gradio
25
  interfaz = gr.Interface(
26
  fn=traducir,
27
  inputs=[
28
  gr.Textbox(lines=5, label="Texto en Español"),
29
+ gr.Dropdown(choices=list(modelos.keys()), label="Traducir a", value="Español a Inglés")
30
  ],
31
  outputs="text",
32
  title="🌍 Traductor Multilingüe",
33
+ description="Traducí textos desde español a varios idiomas con inteligencia artificial 🤖"
34
  )
35
 
36
+ # Lanzar la app
37
  interfaz.launch()