rdlf commited on
Commit
b9925d0
·
verified ·
1 Parent(s): eb9c44b

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +28 -0
app.py ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # app_streamlit_es.py
2
+ import streamlit as st
3
+ from transformers import pipeline
4
+
5
+ # Cargar pipeline genérico de generación
6
+ generador = pipeline("text-generation", model="IIC/MEL") # o modelo instructivo compatible
7
+
8
+ st.set_page_config(page_title="Interface MEL", layout="wide")
9
+ st.title("Interfaz MEL: documentos 🔍")
10
+ st.write("Sube tus documentos, escribe tu prompt (tarea deseada) y obtén la respuesta generada.")
11
+ prompt = st.text_area("Prompt (instrucción al modelo)", height=100, placeholder="Ejemplo: “Resume este texto” o “Clasifica según categoría legal”")
12
+
13
+ archivos = st.file_uploader("Sube documentos (.txt, .md, .pdf)", accept_multiple_files=True)
14
+ if st.button("Generar respuesta"):
15
+ texts = []
16
+ for fichero in archivos:
17
+ try:
18
+ texto = fichero.read().decode("utf‑8")
19
+ except:
20
+ texto = "" # podrías usar pdfplumber o similar para PDF
21
+ texts.append(texto)
22
+ contenido = "\n\n".join(texts)
23
+ entrada = prompt + "\n\n" + contenido
24
+ # Generar con límite de tokens razonable
25
+ salida = generador(entrada, max_new_tokens=512, do_sample=False)
26
+ respuesta = salida[0]["generated_text"]
27
+ st.text_area("Resultado", value=respuesta, height=300)
28
+ st.download_button("Descargar resultado (.txt)", data=respuesta, file_name="resultado.txt")