Spaces:
Running
Running
Update app.py
Browse files
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 = "
|
| 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)
|