Spaces:

iSpr
/

ksic_ai_coding_census2020

Running

iSpr commited on Oct 23, 2022

Commit

4375452

1 Parent(s): df5ecf0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,8 +32,8 @@ def md_loading():
     project_path = './'
     output_model_file = os.path.join(project_path, model_checkpoint)
-    model.load_state_dict(torch.load(output_model_file))
-#    model.load_state_dict(torch.load(output_model_file, map_location=torch.device('cpu')))
 #    ckpt = torch.load(output_model_file, map_location=torch.device('cpu'))
 #    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))
+    model.load_state_dict(torch.load(output_model_file, map_location=torch.device('cpu')))
 #    ckpt = torch.load(output_model_file, map_location=torch.device('cpu'))
 #    model.load_state_dict(ckpt['model_state_dict'])