Update app.py
Browse files
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 |
-
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
|
16 |
-
|
17 |
-
|
|
|
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 |
-
|
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(
|
|
|
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()
|