Spaces:
Sleeping
Sleeping
Chidam Gopal
commited on
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 |
|