basharat8763 commited on
Commit
9cb31ff
·
verified ·
1 Parent(s): f7380de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -8,11 +8,11 @@ from transformers import pipeline
8
  pipe = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6",
9
  torch_dtype=torch.bfloat16)
10
 
11
- # def summary(input):
12
- # output = text_summary(input)
13
- # return output[0]['summary_text']
14
- #
15
- #
16
  gr.close_all()
17
 
18
  # simple gradio web app
 
8
  pipe = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6",
9
  torch_dtype=torch.bfloat16)
10
 
11
+ def summary(input):
12
+ output = text_summary(input)
13
+ return output[0]['summary_text']
14
+
15
+
16
  gr.close_all()
17
 
18
  # simple gradio web app