Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -34,7 +34,7 @@ def get_top95(y_predict, convert_target):
|
|
34 |
return lst_labels
|
35 |
#
|
36 |
# model = MyModel()
|
37 |
-
model = torch.load("model.pt", map_location='cpu')
|
38 |
|
39 |
# print(model)
|
40 |
# model = DistilBertForSequenceClassification.from_pretrained("model/distilbert-model1.pt", local_files_only=True)
|
|
|
34 |
return lst_labels
|
35 |
#
|
36 |
# model = MyModel()
|
37 |
+
model = torch.load("model.pt", map_location='cpu')
|
38 |
|
39 |
# print(model)
|
40 |
# model = DistilBertForSequenceClassification.from_pretrained("model/distilbert-model1.pt", local_files_only=True)
|