Update app.py

app.py

@@ -63,13 +63,15 @@ uploaded_image = st.sidebar.file_uploader("Upload your image here", type=["jpg",
 # Display the uploaded image and process it if available
 if uploaded_image is not None:
     image = Image.open(uploaded_image)
+    if image.mode != 'RGB':
+            image = image.convert('RGB')
     image = image.resize((256,256))
 
     # Save the image to a BytesIO object with a specific format
     image_bytes = io.BytesIO()
     image_format = image.format if image.format else 'PNG'  # Default to 'PNG' if format is None
     image.save(image_bytes, format=image_format)
-    
+    image_bytes.seek(0)
     # Display the image using Streamlit
     st.image(image, caption="Uploaded Image", use_column_width=True)
     image_binary = image_bytes.getvalue()