Ravilppcd07 commited on
Commit
67d9cf7
·
verified ·
1 Parent(s): 650612d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -7,11 +7,11 @@ import os
7
  # Set up Hugging Face API Token (Add your token here)
8
 
9
  # Initialize Hugging Face pipelines
10
- sentiment_pipeline = pipeline("sentiment-analysis", model="allenai/longformer-base-4096", use_auth_token=HUGGING_FACE_API_TOKEN)
11
- classification_pipeline = pipeline("zero-shot-classification", model="facebook/bart-large-mnli", use_auth_token=HUGGING_FACE_API_TOKEN)
12
- ner_pipeline = pipeline("token-classification", model="dslim/bert-base-NER", grouped_entities=True, use_auth_token=HUGGING_FACE_API_TOKEN)
13
- emotion_classifier = pipeline("text-classification", model="bhadresh-savani/distilbert-base-uncased-emotion", use_auth_token=HUGGING_FACE_API_TOKEN)
14
- intent_classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli", use_auth_token=HUGGING_FACE_API_TOKEN)
15
  topic_generator = pipeline("text2text-generation", model="facebook/bart-large-cnn")
16
 
17
  # Define a label mapping
 
7
  # Set up Hugging Face API Token (Add your token here)
8
 
9
  # Initialize Hugging Face pipelines
10
+ sentiment_pipeline = pipeline("sentiment-analysis", model="allenai/longformer-base-4096")
11
+ classification_pipeline = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
12
+ ner_pipeline = pipeline("token-classification", model="dslim/bert-base-NER", grouped_entities=True)
13
+ emotion_classifier = pipeline("text-classification", model="bhadresh-savani/distilbert-base-uncased-emotion")
14
+ intent_classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
15
  topic_generator = pipeline("text2text-generation", model="facebook/bart-large-cnn")
16
 
17
  # Define a label mapping