Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
|
|
4 |
# Initialize pipelines
|
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="
|
8 |
summarization_pipeline = pipeline("summarization", model="facebook/bart-large-cnn")
|
9 |
|
10 |
# Define candidate labels for classification
|
|
|
4 |
# Initialize pipelines
|
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
|