Update app.py
Browse files
app.py
CHANGED
@@ -1,32 +1,39 @@
|
|
1 |
import gradio as gr
|
2 |
from transformers import pipeline
|
|
|
3 |
|
4 |
-
# Pipeline de generación genérica que acepta prompts
|
5 |
generador = pipeline("text-generation", model="IIC/MEL", trust_remote_code=True)
|
6 |
|
7 |
def procesar_tarea(archivos, prompt):
|
8 |
if not prompt:
|
9 |
-
return "", "Debes escribir un prompt"
|
10 |
texto = ""
|
11 |
-
for
|
12 |
try:
|
13 |
-
|
14 |
-
|
15 |
-
|
|
|
|
|
16 |
entrada = prompt + "\n\n" + texto
|
17 |
salida = generador(entrada, max_new_tokens=512, do_sample=False)
|
18 |
respuesta = salida[0].get("generated_text", "")
|
19 |
-
|
|
|
|
|
|
|
|
|
|
|
20 |
|
21 |
demo = gr.Interface(
|
22 |
fn=procesar_tarea,
|
23 |
inputs=[
|
24 |
-
gr.File(label="Sube documentos (.txt, .md, .pdf)", file_count="multiple", type="
|
25 |
gr.Textbox(label="Prompt (instrucción)", lines=3, placeholder="Ej. Resume este texto, explica esto...")
|
26 |
],
|
27 |
outputs=[
|
28 |
gr.Textbox(label="Resultado (para copiar)", lines=10),
|
29 |
-
gr.File(label="Descargar resultado",
|
30 |
],
|
31 |
title="Interfaz MEL con Gradio",
|
32 |
description="Sube tus documentos, escribe un prompt y recibe la salida generada. Puedes copiar el texto o descargarlo."
|
|
|
1 |
import gradio as gr
|
2 |
from transformers import pipeline
|
3 |
+
import tempfile
|
4 |
|
|
|
5 |
generador = pipeline("text-generation", model="IIC/MEL", trust_remote_code=True)
|
6 |
|
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
|
27 |
|
28 |
demo = gr.Interface(
|
29 |
fn=procesar_tarea,
|
30 |
inputs=[
|
31 |
+
gr.File(label="Sube documentos (.txt, .md, .pdf)", file_count="multiple", type="filepath"),
|
32 |
gr.Textbox(label="Prompt (instrucción)", lines=3, placeholder="Ej. Resume este texto, explica esto...")
|
33 |
],
|
34 |
outputs=[
|
35 |
gr.Textbox(label="Resultado (para copiar)", lines=10),
|
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."
|