s2049 commited on
Commit
1ae33d2
·
verified ·
1 Parent(s): 59bd458

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +50 -0
app.py ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+
4
+ def summarize_text(text, min_length, max_length):
5
+ summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
6
+ summary = summarizer(
7
+ text,
8
+ min_length=min_length,
9
+ max_length=max_length,
10
+ do_sample=False
11
+ )
12
+ return summary[0]['summary_text']
13
+
14
+ with gr.Blocks() as demo:
15
+ gr.Markdown("## Adjustable Text Summarization")
16
+ with gr.Row():
17
+ input_text = gr.Textbox(
18
+ label="Enter your text here",
19
+ placeholder="Paste your text here...",
20
+ lines=5
21
+ )
22
+ with gr.Row():
23
+ min_length = gr.Slider(
24
+ label="Minimum Length (Tokens)",
25
+ minimum=10,
26
+ maximum=50,
27
+ value=10
28
+ )
29
+ max_length = gr.Slider(
30
+ label="Maximum Length (Tokens)",
31
+ minimum=50,
32
+ maximum=150,
33
+ value=100
34
+ )
35
+ with gr.Row():
36
+ summarize_button = gr.Button("Generate Summary", variant="primary")
37
+ with gr.Row():
38
+ output_text = gr.Textbox(
39
+ label="Generated Summary",
40
+ lines=4
41
+ )
42
+
43
+ summarize_button.click(
44
+ fn=summarize_text,
45
+ inputs=[input_text, min_length, max_length],
46
+ outputs=output_text
47
+ )
48
+
49
+ if __name__ == "__main__":
50
+ demo.launch()