code fixes

app.py

@@ -96,11 +96,13 @@ if APPLICATION_MODE == "Take Picture":
         st.sidebar.image(picture, caption="captured_picture")
         if st.button("Segment!"):
             ## Function to save image
+
             save_uploadedfile(picture)
             st.sidebar.success("Saved File")
             # captured_picture_img = os.path.join(parent_media_path, "captured_picture.jpg")
+            img_file = os.path.join(parent_media_path, "captured_picture.jpg")
         st.write("Click on **Clear photo** to retake picture")
-        img_file = os.path.join(parent_media_path, "captured_picture.jpg")
+
     st.divider()
 
 elif APPLICATION_MODE == "Upload Picture":
@@ -133,10 +135,12 @@ results = model(img_file)
 img = cv2.imread(img_file)
 names_list = []
 for result in results:
-    boxes = result.boxes.cpu().numpy()
+    # boxes = result.boxes.cpu().numpy()
     # segmentation
-    masks = result.masks.data.cpu().numpy()  # masks, (N, H, W)
-    h, w, _ = img.shape
+    if result.masks:
+        masks = result.masks.data.cpu().numpy()  # masks, (N, H, W)
+    else:
+        masks = []
 
     numCols = len(masks)
     if numCols > 0:
@@ -145,6 +149,7 @@ for result in results:
         print(f"Number of Boxes found: {numCols}")
         st.warning("Unable to id Distinct items - Please retry with a clearer Image")
     for mask in masks:
+        h, w, _ = img.shape
         mask = cv2.resize(mask, (w, h))
         img = overlay(img, mask, color=(0, 255, 0), alpha=0.3)
 #     st.image(rect)