Update app.py

app.py

@@ -126,6 +126,11 @@ def process_image_for_gradio(model, device, image, model_name, threshold=0.4):
     classinfofile = model_configs[model_name]["classinfofile"]
     classinfo = yaml.load(open(classinfofile, "r"), Loader=yaml.FullLoader)["names"]
     indexing_method = "0-based" if "coco" in classinfofile else "1-based"
+
+    print("classinfo: ", classinfo)
+    print()
+    print("indexing_method: ", indexing_method)
+    print()
     
     w, h = im_pil.size
     orig_size = torch.tensor([[w, h]]).to(device)
@@ -141,6 +146,8 @@ def process_image_for_gradio(model, device, image, model_name, threshold=0.4):
     output = model(im_data, orig_size)
     labels, boxes, scores = output
 
+    print("labels: ", labels)
+
     # Visualize results
     detections = sv.Detections(
         xyxy=boxes[0].detach().cpu().numpy(),