Sadem-12 commited on
Commit
720b6ab
·
verified ·
1 Parent(s): 3b8594e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -4,6 +4,12 @@ from transformers import pipeline
4
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
5
 
6
  def summarize_text(text, min_len, max_len):
 
 
 
 
 
 
7
  summary = summarizer(text, min_length=min_len, max_length=max_len)
8
  return summary[0]['summary_text']
9
 
@@ -11,8 +17,8 @@ interface = gr.Interface(
11
  fn=summarize_text,
12
  inputs=[
13
  gr.Textbox(label="Enter Text", lines=10, placeholder="Paste your long text here..."),
14
- gr.Slider(label="Min Length", minimum=10, maximum=50, step=1, value=10), # change default to value
15
- gr.Slider(label="Max Length", minimum=50, maximum=150, step=1, value=100) # change default to value
16
  ],
17
  outputs=gr.Textbox(label="Summarized Text"),
18
  title="Text Summarizer with Sliders",
 
4
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
5
 
6
  def summarize_text(text, min_len, max_len):
7
+
8
+ min_len = max(min_len, 10)
9
+ max_len = min(max_len, 150)
10
+
11
+ min_len = min(min_len, max_len)
12
+
13
  summary = summarizer(text, min_length=min_len, max_length=max_len)
14
  return summary[0]['summary_text']
15
 
 
17
  fn=summarize_text,
18
  inputs=[
19
  gr.Textbox(label="Enter Text", lines=10, placeholder="Paste your long text here..."),
20
+ gr.Slider(label="Min Length", minimum=10, maximum=50, step=1, value=10),
21
+ gr.Slider(label="Max Length", minimum=50, maximum=150, step=1, value=100)
22
  ],
23
  outputs=gr.Textbox(label="Summarized Text"),
24
  title="Text Summarizer with Sliders",