Chidam Gopal commited on
Commit
7538db6
1 Parent(s): 3b08189

app updated with new model

Browse files
Files changed (1) hide show
  1. infer_intent.py +2 -3
infer_intent.py CHANGED
@@ -14,10 +14,9 @@ class IntentClassifier:
14
  7: 'unknown'}
15
  self.label2id = {label:id for id,label in self.id2label.items()}
16
 
17
- self.tokenizer = AutoTokenizer.from_pretrained("chidamnat2002/intent_classifier")
18
- self.intent_model = AutoModelForSequenceClassification.from_pretrained('chidamnat2002/intent_classifier',
19
  num_labels=8,
20
- torch_dtype=torch.bfloat16,
21
  id2label=self.id2label,
22
  label2id=self.label2id)
23
 
 
14
  7: 'unknown'}
15
  self.label2id = {label:id for id,label in self.id2label.items()}
16
 
17
+ self.tokenizer = AutoTokenizer.from_pretrained("Mozilla/mobilebert-uncased-finetuned-LoRA-intent-classifier")
18
+ self.intent_model = AutoModelForSequenceClassification.from_pretrained('Mozilla/mobilebert-uncased-finetuned-LoRA-intent-classifier',
19
  num_labels=8,
 
20
  id2label=self.id2label,
21
  label2id=self.label2id)
22