update

app.py

@@ -18,7 +18,7 @@ if torch.cuda.is_available():
     net=net.cuda()
     device = "cuda"
 elif torch.backends.mps.is_available():
-    net.load_state_dict(torch.load(model_path))
+    net.load_state_dict(torch.load(model_path,map_location="mps"))
     net=net.to("mps")
     device = "mps"
 else:
@@ -45,10 +45,9 @@ def process(image):
     im_tensor = torch.divide(im_tensor,255.0)
     im_tensor = normalize(im_tensor,[0.5,0.5,0.5],[1.0,1.0,1.0])
     if device == "cuda":
+        im_tensor=im_tensor.cuda()
     elif device == "mps":
         im_tensor=im_tensor.to("mps")
-    if torch.cuda.is_available():
-        im_tensor=im_tensor.cuda()
 
     #inference
     result=net(im_tensor)