Spaces:
Running
Running
a96123155
commited on
Commit
·
85f2ad4
1
Parent(s):
3ca49d5
app
Browse files
app.py
CHANGED
@@ -475,8 +475,8 @@ def read_raw(raw_input):
|
|
475 |
|
476 |
def predict_raw(raw_input):
|
477 |
modelfile = 'v2.7_LeidenContrastive_best_model_fold0.pt'
|
478 |
-
|
479 |
-
|
480 |
|
481 |
for k, v in state_dict.items():
|
482 |
name = k.replace('module.','')
|
|
|
475 |
|
476 |
def predict_raw(raw_input):
|
477 |
modelfile = 'v2.7_LeidenContrastive_best_model_fold0.pt'
|
478 |
+
state_dict = torch.load(modelfile, map_location=torch.device(device))
|
479 |
+
new_state_dict = OrderedDict()
|
480 |
|
481 |
for k, v in state_dict.items():
|
482 |
name = k.replace('module.','')
|