Update app.py

app.py

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