a96123155 commited on
Commit
85f2ad4
·
1 Parent(s): 3ca49d5
Files changed (1) hide show
  1. app.py +2 -2
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
- 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.','')
 
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.','')