rdlf commited on
Commit
82366d1
·
verified ·
1 Parent(s): 219595f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -11
app.py CHANGED
@@ -7,20 +7,21 @@ generador = pipeline("text-generation", model="IIC/MEL", trust_remote_code=True)
7
  def procesar_tarea(archivos, prompt):
8
  if not prompt:
9
  return "", None, "Debes escribir un prompt"
 
10
  texto = ""
11
- for fpath in archivos:
12
- try:
13
- with open(fpath, "r", encoding="utf‑8") as f:
14
- texto += f.read() + "\n\n"
15
- except Exception:
16
- # Ignorar si no se puede leer
17
- pass
 
18
  entrada = prompt + "\n\n" + texto
19
  salida = generador(entrada, max_new_tokens=512, do_sample=False)
20
  respuesta = salida[0].get("generated_text", "")
21
 
22
- # Guardar en un archivo temporal para descarga
23
- tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".txt", mode="w", encoding="utf‑8")
24
  tmp.write(respuesta)
25
  tmp.close()
26
  return respuesta, tmp.name, None
@@ -36,8 +37,8 @@ demo = gr.Interface(
36
  gr.File(label="Descargar resultado", interactive=False)
37
  ],
38
  title="Interfaz MEL con Gradio",
39
- description="Sube tus documentos, escribe un prompt y recibe la salida generada. Puedes copiar el texto o descargarlo."
40
  )
41
 
42
  if __name__ == "__main__":
43
- demo.launch(share=True)
 
7
  def procesar_tarea(archivos, prompt):
8
  if not prompt:
9
  return "", None, "Debes escribir un prompt"
10
+
11
  texto = ""
12
+ if archivos:
13
+ for fpath in archivos:
14
+ try:
15
+ with open(fpath, "r", encoding="utf-8") as f:
16
+ texto += f.read() + "\n\n"
17
+ except Exception:
18
+ pass # evita fallos si no se puede leer algún archivo
19
+
20
  entrada = prompt + "\n\n" + texto
21
  salida = generador(entrada, max_new_tokens=512, do_sample=False)
22
  respuesta = salida[0].get("generated_text", "")
23
 
24
+ tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".txt", mode="w", encoding="utf-8")
 
25
  tmp.write(respuesta)
26
  tmp.close()
27
  return respuesta, tmp.name, None
 
37
  gr.File(label="Descargar resultado", interactive=False)
38
  ],
39
  title="Interfaz MEL con Gradio",
40
+ description="Sube tus documentos (opcional), escribe un prompt y recibe la salida generada. Puedes copiar el texto o descargarlo."
41
  )
42
 
43
  if __name__ == "__main__":
44
+ demo.launch()