Spaces:

iSpr
/

ksic_ai_coding_census2020

Runtime error

iSpr commited on Oct 14, 2022

Commit

d689387

1 Parent(s): b263e68

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -23,8 +23,9 @@ def md_loading():
     model_checkpoint = 'base1_43_11.bin'
     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')

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