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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="elastic/distilbert-base-uncased-finetuned-conll03-english", aggregation_strategy="simple")
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