改成cpu运行

app.py

@@ -419,7 +419,7 @@ def merge_images(images: np.ndarray):
 
 def get_models(device):
     def modelLoad(model, model_path, data_parallel=False):
-        model.load_state_dict(torch.load(model_path), strict=True)
+        model.load_state_dict(torch.load(model_path, map_location='cpu'), strict=True)
 
         if data_parallel:
             model = torch.nn.DataParallel(model)