Spaces:
Running
Running
Chidam Gopal
commited on
Commit
•
7538db6
1
Parent(s):
3b08189
app updated with new model
Browse files- 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("
|
18 |
-
self.intent_model = AutoModelForSequenceClassification.from_pretrained('
|
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 |
|