Curative commited on
Commit
7856ffd
·
verified ·
1 Parent(s): ceedc97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -11
app.py CHANGED
@@ -1,22 +1,24 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- # Load the summarization model
5
  summarizer = pipeline("summarization", model="Curative/t5-summarizer-cnn")
6
 
7
- # Define the summarization function
8
  def summarize_text(input_text):
9
- summary = summarizer(input_text, max_length=150, min_length=30, do_sample=False)
10
- return summary[0]['summary_text']
 
 
11
 
12
- # Create the Gradio UI
13
- iface = gr.Interface(
14
  fn=summarize_text,
15
- inputs=gr.Textbox(lines=5, placeholder="Enter text to summarize..."),
16
  outputs="text",
17
- title="T5 Text Summarizer",
18
- description="Enter a passage, and the T5 model will generate a concise summary."
19
  )
20
 
21
- # Launch the app
22
- iface.launch()
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ # Load your summarization model
5
  summarizer = pipeline("summarization", model="Curative/t5-summarizer-cnn")
6
 
7
+ # Define function for Gradio
8
  def summarize_text(input_text):
9
+ if not input_text.strip():
10
+ return "Please enter some text to summarize."
11
+ result = summarizer(input_text, max_length=150, min_length=40, do_sample=False)
12
+ return result[0]['summary_text']
13
 
14
+ # Create Gradio interface
15
+ demo = gr.Interface(
16
  fn=summarize_text,
17
+ inputs=gr.Textbox(lines=10, placeholder="Enter text to summarize..."),
18
  outputs="text",
19
+ title="T5 Summarizer (CNN/DailyMail)",
20
+ description="A fine-tuned T5 model for text summarization (based on cnn_dailymail v3.0.0)."
21
  )
22
 
23
+ # Launch app
24
+ demo.launch()