Sadem-12 commited on
Commit
cb38f8c
·
verified ·
1 Parent(s): 93a2ed0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -9
app.py CHANGED
@@ -1,17 +1,11 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- try:
5
- summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
6
- except Exception as e:
7
- print(f"Error loading model: {e}")
8
 
9
  def summarize_text(text):
10
- try:
11
- summary = summarizer(text, min_length=10, max_length=100)
12
- return summary[0]['summary_text']
13
- except Exception as e:
14
- return f"Error during summarization: {e}"
15
 
16
  interface = gr.Interface(
17
  fn=summarize_text,
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
 
 
5
 
6
  def summarize_text(text):
7
+ summary = summarizer(text, min_length=10, max_length=100)
8
+ return summary[0]['summary_text']
 
 
 
9
 
10
  interface = gr.Interface(
11
  fn=summarize_text,