rasulbrur commited on
Commit
ebe592c
·
verified ·
1 Parent(s): 72801e6

Update voice/intent_classifier.py

Browse files
Files changed (1) hide show
  1. voice/intent_classifier.py +1 -1
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",