Update app.py

app.py

@@ -76,6 +76,10 @@ def detect_objects_in_image(image):
 
     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
+
     mask = pred[:, 4] > conf_thres
     pred = pred[mask]