Spaces:

iSpr
/

ksic_ai_coding_census2020

Runtime error

iSpr commited on Oct 14, 2022

Commit

b263e68

1 Parent(s): 1e967b6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,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, map_location=torch.device('cpu')))
 ################################## label tbl 수정
     label_tbl = np.load('./label_table.npy')

     project_path = './'
     output_model_file = os.path.join(project_path, model_checkpoint)
+    model.load_state_dict(torch.load(output_model_file['model_state_dict'], map_location=torch.device('cpu')))
 ################################## label tbl 수정
     label_tbl = np.load('./label_table.npy')