feat

app.py

@@ -243,8 +243,6 @@ def count_objects(image, pos_caption, neg_caption, box_threshold, point_radius,
     neg_queries = outputs["neg_queries"].squeeze(0).float()
     pos_queries = pos_queries[-1].squeeze(0)
     neg_queries = neg_queries[-1].squeeze(0)
-    pos_queries = pos_queries.cpu()
-    neg_queries = neg_queries.cpu()
     pos_queries = pos_queries.unsqueeze(0)
     neg_queries = neg_queries.unsqueeze(0)
     results = post_process_grounded_object_detection_with_queries(outputs, pos_queries, box_threshold=threshold)[0]
@@ -268,8 +266,8 @@ def count_objects(image, pos_caption, neg_caption, box_threshold, point_radius,
     
     pos_queries = results["queries"]
     neg_queries = neg_results["queries"]
-    pos_queries = pos_queries.numpy()
-    neg_queries = neg_queries.numpy()
+    pos_queries = pos_queries.cpu().numpy()
+    neg_queries = neg_queries.cpu().numpy()
     
     img_size = image.size
     # filtered_points, kept_indices = filter_points_by_negative(