Update app.py

app.py

@@ -58,20 +58,22 @@ def detect_objects_in_image(image):
         image = Image.fromarray(image)
     print(image.size)
     orig_w, orig_h = image.size
+    print("passed1")
 
 
     with torch.no_grad():
         pred = model(img_tensor)[0]
+    print("Passed2")
 
     if isinstance(pred[0], torch.Tensor):
         pred = [p.cpu().numpy() for p in pred]
-
+    print("Passed3")
     pred = np.concatenate(pred, axis=0)
     conf_thres = 0.25
 
         # Ensure `pred` is at least a 2D array before indexing
     pred = np.atleast_2d(pred)  # Converts 1D to 2D if necessary
-
+    print("passed4")
     mask = pred[:, 4] > conf_thres
     pred = pred[mask]