Update app.py

app.py

@@ -82,16 +82,19 @@ def fix_image(upload=None):
     
     image = Image.open(upload)
     
-    # Display the original image
-    st.image(image, caption="Uploaded Image", use_column_width=True)
-
     # Prepare image for segmentation
     image_batch = prepare_single_image(image, resize_shape, transforms, "cpu")
     prediction_u2net = prepare_prediction(u2net, image_batch)
     masked_image = apply_mask(image, prediction_u2net)
-
-    # Display segmented image
-    st.image(masked_image, caption='Segmented Image', use_column_width=True)
+    
+    # Display the original and segmented images side by side
+    col1, col2 = st.columns(2)
+    
+    with col1:
+        st.image(image, caption="Uploaded Image", use_column_width=True)
+    
+    with col2:
+        st.image(masked_image, caption='Segmented Image', use_column_width=True)
 
     # Provide download option for segmented image
     buf = io.BytesIO()