Curative commited on
Commit
d588f49
·
verified ·
1 Parent(s): b0c378e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="facebook/bart-large-cnn")
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 = [