Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ from transformers import pipeline
|
|
5 |
sentiment_pipeline = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
|
6 |
classification_pipeline = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
|
7 |
ner_pipeline = pipeline("ner", model="dslim/bert-base-NER", aggregation_strategy="simple")
|
8 |
-
summarization_pipeline = pipeline("summarization", model="
|
9 |
|
10 |
# Define candidate labels for classification
|
11 |
candidate_labels = [
|
|
|
5 |
sentiment_pipeline = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
|
6 |
classification_pipeline = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
|
7 |
ner_pipeline = pipeline("ner", model="dslim/bert-base-NER", aggregation_strategy="simple")
|
8 |
+
summarization_pipeline = pipeline("summarization", model="Curative/t5-summarizer-cnn")
|
9 |
|
10 |
# Define candidate labels for classification
|
11 |
candidate_labels = [
|