iabd10 commited on
Commit
4f25a39
·
verified ·
1 Parent(s): 221729e

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +28 -0
app.py ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
2
+ import gradio as gr
3
+
4
+ # Cargar el tokenizer y el modelo desde el repositorio del modelo preentrenado
5
+ tokenizer = AutoTokenizer.from_pretrained("mrm8488/bert2bert_shared-spanish-finetuned-summarization")
6
+ model = AutoModelForSeq2SeqLM.from_pretrained("mrm8488/bert2bert_shared-spanish-finetuned-summarization")
7
+
8
+ # Crear el pipeline de resumen utilizando los objetos cargados
9
+ resumidor = pipeline("summarization", model=model, tokenizer=tokenizer)
10
+
11
+ def generar_resumen(texto):
12
+ # Generar el resumen usando el pipeline
13
+ resumen = resumidor(texto, max_length=130, min_length=30, do_sample=False)
14
+ return resumen[0]["summary_text"]
15
+
16
+ # Crear la interfaz de Gradio
17
+ demo = gr.Interface(
18
+ fn=generar_resumen,
19
+ inputs=gr.Textbox(lines=10, placeholder="Pega aquí un texto largo en español..."),
20
+ outputs="text",
21
+ title="Generación de Resúmenes Automáticos",
22
+ description="Introduce un texto largo en español y obtén un resumen generado automáticamente por un modelo preentrenado."
23
+ )
24
+
25
+ # Lanzar la aplicación
26
+ demo.launch()
27
+
28
+