Spaces:

iSpr
/

ksic_ai_coding_census2020

Runtime error

iSpr commited on Oct 23, 2022

Commit

cc5a679

1 Parent(s): 876124e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -31,7 +31,7 @@ def md_loading():
     project_path = './'
     output_model_file = os.path.join(project_path, model_checkpoint)
-    model.load_state_dict(torch.load(output_model_file))
 #    ckpt = torch.load(output_model_file)
 #    model.load_state_dict(ckpt['model_state_dict'])

     project_path = './'
     output_model_file = os.path.join(project_path, model_checkpoint)
+    model.load_state_dict(torch.load(output_model_file, map_location=torch.device('cpu')))
 #    ckpt = torch.load(output_model_file)
 #    model.load_state_dict(ckpt['model_state_dict'])