update app.py

app.py

@@ -60,7 +60,7 @@ def load_model_hf(model_config_path, repo_id, filename, device='cpu'):
     args.device = device
 
     cache_file = hf_hub_download(repo_id=repo_id, filename=filename)
-    checkpoint = torch.load(cache_file, map_location='cpu')
+    checkpoint = torch.load(cache_file, map_location=device)
     log = model.load_state_dict(clean_state_dict(checkpoint['model']), strict=False)
     print("Model loaded from {} \n => {}".format(cache_file, log))
     _ = model.eval()
@@ -227,6 +227,7 @@ def run_grounded_sam(image_path, text_prompt, task_type, inpaint_prompt, box_thr
         except:
             warnings.warn("Failed to load custom C++ ops. Running on CPU mode Only in groundingdino!")
 
+    groundingdino_device = 'cpu'
     boxes_filt, pred_phrases = get_grounding_output(
         groundingdino_model, image, text_prompt, box_threshold, text_threshold, device=groundingdino_device
     )