Spaces:

iSpr
/

ksic_ai_coding_census2020

Runtime error

iSpr commited on Oct 14, 2022

Commit

694f95f

1 Parent(s): d689387

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ def md_loading():
     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')

     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')