kubrabuzlu commited on
Commit
ae85e9b
·
verified ·
1 Parent(s): 89c2cc7

perf: Optimized analyze_text function for faster execution

Browse files
src/SentimentAndIntentionAnalysis.py CHANGED
@@ -6,6 +6,7 @@ class ZeroShotClassifier:
6
  self.model_name = model_name
7
  self.sentiment_labels = ["Positive", "Negative", "Neutral"]
8
  self.intention_labels = ["Inquire", "Inform", "Payment", "Price", "Trade In", "Discount", "Complaint", "Approve", "Selling", "Confusion", "Change Package", "Upgrade", "Purchase", "Help"]
 
9
 
10
  def create_model(self, model_name):
11
  # Create Model
 
6
  self.model_name = model_name
7
  self.sentiment_labels = ["Positive", "Negative", "Neutral"]
8
  self.intention_labels = ["Inquire", "Inform", "Payment", "Price", "Trade In", "Discount", "Complaint", "Approve", "Selling", "Confusion", "Change Package", "Upgrade", "Purchase", "Help"]
9
+ self.labels = self.sentiment_labels + self.intention_labels
10
 
11
  def create_model(self, model_name):
12
  # Create Model