zoya23 commited on
Commit
b40112c
·
verified ·
1 Parent(s): ed360a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def load_examples():
19
  examples = load_examples()
20
 
21
  # Set up the summarization model
22
- summarizer = pipeline("summarization", model="t5-small")
23
 
24
  # Few-shot prompt template
25
  example_prompt = FewShotChatMessagePromptTemplate.from_examples(
 
19
  examples = load_examples()
20
 
21
  # Set up the summarization model
22
+ summarizer = pipeline("summarization", model="google/pegasus-xsum")
23
 
24
  # Few-shot prompt template
25
  example_prompt = FewShotChatMessagePromptTemplate.from_examples(