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

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
- 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):
 
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):