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