abdull4h commited on
Commit
3315f36
·
verified ·
1 Parent(s): d159523

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +51 -0
app.py ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+
4
+ # Initialize the summarization pipeline with facebook/bart-large-cnn
5
+ summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
6
+
7
+ def summarize_text(text, min_len, max_len):
8
+ """
9
+ Summarize the input text using the specified min_len and max_len.
10
+ """
11
+ # Convert slider values to integers
12
+ min_len = int(min_len)
13
+ max_len = int(max_len)
14
+
15
+ # Run the summarizer with the given lengths
16
+ summary = summarizer(text, min_length=min_len, max_length=max_len)
17
+ return summary[0]['summary_text']
18
+
19
+ # Build the Gradio interface
20
+ demo = gr.Interface(
21
+ fn=summarize_text,
22
+ inputs=[
23
+ gr.Textbox(
24
+ lines=10,
25
+ placeholder="Enter a long piece of text here...",
26
+ label="Input Text"
27
+ ),
28
+ gr.Slider(
29
+ minimum=10,
30
+ maximum=50,
31
+ step=1,
32
+ value=25,
33
+ label="Minimum Summary Length (tokens)"
34
+ ),
35
+ gr.Slider(
36
+ minimum=50,
37
+ maximum=150,
38
+ step=1,
39
+ value=100,
40
+ label="Maximum Summary Length (tokens)"
41
+ )
42
+ ],
43
+ outputs=gr.Textbox(
44
+ label="Summary"
45
+ ),
46
+ title="BART Text Summarizer with Adjustable Lengths",
47
+ description="Enter a long piece of text, adjust the summary length settings using the sliders, and click 'Submit' to generate a summary."
48
+ )
49
+
50
+ if __name__ == "__main__":
51
+ demo.launch()