Update app.py

app.py

@@ -12,11 +12,10 @@ from segment_anything import SamAutomaticMaskGenerator, SamPredictor, sam_model_
 
 matplotlib.pyplot.switch_backend('Agg') # for matplotlib to work in gradio
 
-device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') # use GPU if available
 
 #setup model
 sam_checkpoint = "sam_vit_h_4b8939.pth"
-device = "cuda"
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') # use GPU if available
 model_type = "default"
 sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)
 sam.to(device=device)