Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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"
|
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
|
|
|
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
|