dev2004v commited on
Commit
db17d73
·
verified ·
1 Parent(s): d6cc012

Update app/core/summarizer.py

Browse files
Files changed (1) hide show
  1. app/core/summarizer.py +1 -1
app/core/summarizer.py CHANGED
@@ -28,7 +28,7 @@ def get_summary(text):
28
  raise ValueError("Text too long to summarize. Please try a shorter page.")
29
 
30
  from transformers import pipeline
31
- summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
32
 
33
  summaries = []
34
  chunks = chunk_text(text)
 
28
  raise ValueError("Text too long to summarize. Please try a shorter page.")
29
 
30
  from transformers import pipeline
31
+ summarizer = pipeline("summarization", model="philschmid/bart-large-cnn-samsum")
32
 
33
  summaries = []
34
  chunks = chunk_text(text)