update app.py

app.py

@@ -185,6 +185,8 @@ device = "cuda"
 
 device = get_device()
 
+print(f'device={device}')
+
 # initialize groundingdino model
 groundingdino_model = load_model_hf(config_file, ckpt_repo_id, ckpt_filenmae)
 
@@ -194,7 +196,7 @@ sam_predictor = SamPredictor(build_sam(checkpoint=sam_checkpoint))
 # initialize stable-diffusion-inpainting
 sd_pipe = StableDiffusionInpaintPipeline.from_pretrained(
         "runwayml/stable-diffusion-inpainting", 
-        # torch_dtype=torch.float16
+        torch_dtype=torch.float16
 )
 sd_pipe = sd_pipe.to(device)
 
@@ -211,7 +213,7 @@ def run_grounded_sam(image_path, text_prompt, task_type, inpaint_prompt, box_thr
 
     # run grounding dino model
     boxes_filt, pred_phrases = get_grounding_output(
-        groundingdino_model, image, text_prompt, box_threshold, text_threshold, device=device
+        groundingdino_model, image, text_prompt, box_threshold, text_threshold, device='cpu'
     )
 
     size = image_pil.size