Update app.py

app.py

@@ -12,7 +12,12 @@ from PIL import Image
 from segment_anything_2 import SAM2ImagePredictor, build_sam2
 
 # Load your model
-device = "cuda" if torch.cuda.is_available() else "cpu"
+import torch
+
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model = model.to(device)
+
+#device = "cuda" if torch.cuda.is_available() else "cpu"
 checkpoint = "checkpoints/sam2_hiera_large.pt"
 model_cfg = "sam2_hiera_l.yaml"
 model = build_sam2(model_cfg, checkpoint, device=device)