GuidK commited on
Commit
d4fd203
·
verified ·
1 Parent(s): 9e6c9cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import html as html_lib
6
  classifier = pipeline("sentiment-analysis")
7
 
8
  # Set model name for summarization
9
- sum_model_name = "facebook/bart-large-cnn"
10
 
11
  # Initialize tokenizer for summarization model
12
  sum_tokenizer = AutoTokenizer.from_pretrained(sum_model_name)
 
6
  classifier = pipeline("sentiment-analysis")
7
 
8
  # Set model name for summarization
9
+ sum_model_name = "sshleifer/distilbart-cnn-12-6"
10
 
11
  # Initialize tokenizer for summarization model
12
  sum_tokenizer = AutoTokenizer.from_pretrained(sum_model_name)