Commit
•
c6682ca
1
Parent(s):
ae3ff72
Update app.py
Browse files
app.py
CHANGED
|
@@ -45,7 +45,7 @@ tokenizer = RobertaTokenizer.from_pretrained(BERT_MODEL_NAME_FOR_CATEGORIES_CLAS
|
|
| 45 |
LABEL_COLUMNS_CATEGORIES = ['AMBIENCE', 'DRINK', 'FOOD', 'GENERAL', 'RESTAURANT', 'SERVICE', 'STAFF']
|
| 46 |
|
| 47 |
categories_model = RobertaForSequenceClassification.from_pretrained(BERT_MODEL_NAME_FOR_CATEGORIES_CLASSIFICATION, num_labels=len(LABEL_COLUMNS_CATEGORIES))
|
| 48 |
-
categories_model.load_state_dict(torch.load('./Categories_Classification_Model_updated.pth'))
|
| 49 |
categories_model.eval()
|
| 50 |
|
| 51 |
# Streamlit App
|
|
|
|
| 45 |
LABEL_COLUMNS_CATEGORIES = ['AMBIENCE', 'DRINK', 'FOOD', 'GENERAL', 'RESTAURANT', 'SERVICE', 'STAFF']
|
| 46 |
|
| 47 |
categories_model = RobertaForSequenceClassification.from_pretrained(BERT_MODEL_NAME_FOR_CATEGORIES_CLASSIFICATION, num_labels=len(LABEL_COLUMNS_CATEGORIES))
|
| 48 |
+
categories_model.load_state_dict(torch.load('./Categories_Classification_Model_updated.pth',map_location=torch.device('cpu') ))
|
| 49 |
categories_model.eval()
|
| 50 |
|
| 51 |
# Streamlit App
|