Spaces:
Running
Running
Update voice/intent_classifier.py
Browse files
voice/intent_classifier.py
CHANGED
@@ -10,7 +10,7 @@ class IntentClassifier:
|
|
10 |
def __init__(self):
|
11 |
# Use a larger model for better NER (optional)
|
12 |
self.nlp = spacy.load("en_core_web_sm") # "en_core_web_lg"
|
13 |
-
self.classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
|
14 |
self.intents = [
|
15 |
"get_net_income",
|
16 |
"get_revenue",
|
|
|
10 |
def __init__(self):
|
11 |
# Use a larger model for better NER (optional)
|
12 |
self.nlp = spacy.load("en_core_web_sm") # "en_core_web_lg"
|
13 |
+
self.classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli", from_pt=True)
|
14 |
self.intents = [
|
15 |
"get_net_income",
|
16 |
"get_revenue",
|