Update app.py

app.py

@@ -33,9 +33,9 @@ def load_model(selected_model, model):
     global model_path
     model_path = os.path.join(model_folder_path, selected_model)
     logger.info(f'选择模型路径为{model_path}')
-    state_dict = torch.load(model_path)
+    state_dict = torch.load(model_path,map_location = torch.device('cpu'))
     logger.info(f"加载的state_dict大小: {sum([param.numel() for param in state_dict.values()])}")
-    model.load_state_dict(state_dict, map_location = torch.device('cpu'))
+    model.load_state_dict(state_dict)
     model.eval()
     model.to(DEVICE)