Update app.py

app.py

@@ -148,7 +148,7 @@ def inference(image_path, model_name, bbox_threshold):
     
     ort_inputs = {ort_session.get_inputs()[0].name: tensor_img.cpu().numpy()}
     pred_boxes, logits = ort_session.run(['dets', 'labels'], ort_inputs)
-    
+    print(pred_boxes)
     scores = torch.sigmoid(torch.from_numpy(logits))
     max_scores, pred_labels = scores.max(-1)
     mask = max_scores > bbox_threshold
@@ -171,7 +171,7 @@ def inference(image_path, model_name, bbox_threshold):
     filtered_labels = pred_labels.squeeze(0)[mask]
     
     img_tensor = torch.from_numpy(np.array(image)).permute(2, 0, 1)
-    
+    print("drawing")
     return draw_predictions(filtered_boxes, filtered_labels, filtered_scores, img_tensor, score_threshold=bbox_threshold)