bickallen commited on
Commit
40d1ff8
·
verified ·
1 Parent(s): 265749e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -5,11 +5,10 @@ from transformers import pipeline
5
 
6
  # Create a summarization 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="auto")
11
- #text_summery = pipeline("summarization", model="google/pegasus-xsum", torch_dtype=torch.bfloat16)
12
- text_summery = pipeline("summarization", model="prithivMLmods/t5-Flan-Prompt-Enhance")
13
 
14
  # Define a function that takes a text and returns a summary
15
  def summarize(text):
 
5
 
6
  # Create a summarization pipeline
7
  # summarizer = pipeline("summarization")
8
+ #text_summery = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
9
+ text_summery = pipeline("summarization", model="sshleifer/distilbart-cnn-6-6")
10
+ #text_summery = pipeline("summarization", model="facebook/bart-large-cnn")
11
+ #text_summery = pipeline("summarization", model="google/pegasus-xsum")
 
12
 
13
  # Define a function that takes a text and returns a summary
14
  def summarize(text):