Update app.py

app.py

@@ -38,7 +38,7 @@ def predict(img_name, model=None, device=None, reduce_labels=True):
             pre_img_data[:,:,i] = normalize_channel(img_channel_i, lower=1, upper=99)
 
     my_model = MultiStreamCellSegModel.from_pretrained("Lewislou/cellseg_sribd")
-    checkpoints = torch.load('model.pt')
+    checkpoints = torch.load('model.pt',map_location=torch.device('cpu'))
     my_model.__init__(ModelConfig())
     my_model.load_checkpoints(checkpoints)
     with torch.no_grad():