Update app.py

app.py

@@ -16,7 +16,7 @@ def set_predictor(image):
     """
     Creates a Sam predictor object based on a given image and model.
     """
-    device = 'cpu'
+    device = 'cuda' if torch.cuda.is_available() else 'cpu'
     sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)
     sam.to(device=device)