Update app.py

app.py

@@ -31,9 +31,9 @@ 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')))
-#    ckpt = torch.load(output_model_file)
-#    model.load_state_dict(ckpt['model_state_dict'])
+#    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'])
     
 #    device = torch.device("cuda" if torch.cuda.is_available() and not False else "cpu")
 #    device = torch.device("cpu")