update app.py

app.py

@@ -242,7 +242,8 @@ groundingdino_model = load_model_hf(config_file, ckpt_repo_id, ckpt_filenmae)
 
 # initialize SAM
 logger.info(f"initialize SAM model...")
-sam_model = build_sam(checkpoint=sam_checkpoint).to(device)
+sam_device = device
+sam_model = build_sam(checkpoint=sam_checkpoint).to(sam_device)
 sam_predictor = SamPredictor(sam_model)
 sam_mask_generator = SamAutomaticMaskGenerator(sam_model)
 
@@ -558,7 +559,7 @@ def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_t
             boxes_filt[i][:2] -= boxes_filt[i][2:] / 2
             boxes_filt[i][2:] += boxes_filt[i][:2]
 
-        boxes_filt = boxes_filt.cpu()
+        boxes_filt = boxes_filt.to(sam_device)
         transformed_boxes = sam_predictor.transform.apply_boxes_torch(boxes_filt, image.shape[:2])
 
         masks, _, _, _ = sam_predictor.predict_torch(