bickallen commited on
Commit
f48b442
·
verified ·
1 Parent(s): ee70c41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,8 +7,8 @@ from transformers import pipeline
7
  # summarizer = pipeline("summarization")
8
  #text_summery = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
9
  #text_summery = pipeline("summarization", model="sshleifer/distilbart-cnn-6-6", torch_dtype=torch.bfloat16)
10
- text_summery = pipeline("summarization", model="facebook/bart-large-cnn", torch_dtype=torch.bfloat16)
11
-
12
 
13
  # Define a function that takes a text and returns a summary
14
  def summarize(text):
 
7
  # summarizer = pipeline("summarization")
8
  #text_summery = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
9
  #text_summery = pipeline("summarization", model="sshleifer/distilbart-cnn-6-6", torch_dtype=torch.bfloat16)
10
+ #text_summery = pipeline("summarization", model="facebook/bart-large-cnn", torch_dtype=torch.bfloat16)
11
+ text_summery = pipeline("summarization", model="google/pegasus-xsum", torch_dtype=torch.bfloat16)
12
 
13
  # Define a function that takes a text and returns a summary
14
  def summarize(text):