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()
|