Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,10 +2,15 @@ import streamlit as st
 from PIL import Image
 import io
 
-st.title("Minimal Image Upload Test")
+# Display logo (make sure the image is inside your repo, path correct)
+st.image("src/RezbinLogo.jpg", width=150)
 
-uploaded_file = st.file_uploader("Upload image", type=["png", "jpg", "jpeg"])
+st.title("Rezbin AI 📷")
+st.write("Upload an image of your waste before throwing it.")
+
+uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png", "jpeg"])
 
 if uploaded_file is not None:
-    img = Image.open(io.BytesIO(uploaded_file.read()))
-    st.image(img, caption="Uploaded Image")
+    # Read the uploaded image file directly into PIL Image
+    img = Image.open(uploaded_file)
+    st.image(img, caption="Uploaded Image", use_container_width=True)