Update app.py

app.py

@@ -96,8 +96,9 @@ def main():
     segmentation_mask_no_prompt = segment_with_medsam(tiff_image, mask_np, False)
     segmentation_mask_with_prompt = segment_with_medsam(tiff_image, mask_np, True)
     st.subheader("Segmentation Results")
-    st.image(segmentation_mask_no_prompt, caption="Segmented Image without prompt")
-    st.image(segmentation_mask_with_prompt, caption="Segmented Image with prompt")
+    st.image(tiff_image, caption="Uploaded Image")
+    st.image(segmentation_mask_no_prompt, caption="Segmented Image")
+    st.image(segmentation_mask_with_prompt, caption="Segmented Image with occlusion fixed")
 
 if __name__ == "__main__":
   main()