Update app.py

app.py

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