Update app.py

app.py

@@ -76,11 +76,10 @@ uploaded_file = st.sidebar.file_uploader("Choose an image...", type=["png", "jpg
 
 # Function to handle image and segmentation display
 def fix_image(upload=None):
-    if upload is None:
-        st.write("Please upload an image.")
-        return
-    
-    image = Image.open(upload)
+    if upload:
+        image = Image.open(upload)
+    else:
+        image = Image.open("8.jpg")
     
     # Prepare image for segmentation
     image_batch = prepare_single_image(image, resize_shape, transforms, "cpu")
@@ -108,6 +107,8 @@ def fix_image(upload=None):
         mime="image/png"
     )
 
-# Handle image processing based on user input
+
 if uploaded_file is not None:
     fix_image(upload=uploaded_file)
+else:
+    fix_image()  # Use default image if none uploaded