printing devices

app.py

@@ -231,10 +231,15 @@ def count(image, text, prompts, state, device):
     input_image, _ = transform(image, {"exemplars": torch.tensor([])})
     input_image = input_image.unsqueeze(0).to(device)
     exemplars = get_box_inputs(prompts["points"])
-    print(exemplars)
+    
     input_image_exemplars, exemplars = transform(prompts["image"], {"exemplars": torch.tensor(exemplars)})
     input_image_exemplars = input_image_exemplars.unsqueeze(0).to(device)
     exemplars = [exemplars["exemplars"].to(device)]
+
+    print("model device: " + str(model.device))
+    print("input image device: " + str(input_image.device))
+    print("input image exemplars device: " + str(input_image_exemplars.device))
+    print("exemplars device: " + str(exemplars[0].device))
     
     with torch.no_grad():
         model_output = model(
@@ -316,7 +321,7 @@ def count_main(image, text, prompts, device):
     input_image, _ = transform(image, {"exemplars": torch.tensor([])})
     input_image = input_image.unsqueeze(0).to(device)
     exemplars = get_box_inputs(prompts["points"])
-    print(exemplars)
+    
     input_image_exemplars, exemplars = transform(prompts["image"], {"exemplars": torch.tensor(exemplars)})
     input_image_exemplars = input_image_exemplars.unsqueeze(0).to(device)
     exemplars = [exemplars["exemplars"].to(device)]