Update app.py

app.py

@@ -1,9 +1,7 @@
 from PIL import Image
 import numpy as np
-import streamlit as st
-
-from PIL import Image
-import numpy as np
+import io
+from huggingface_hub import hf_shiny as hf
 
 # Define function to process uploaded image
 def process_image(image):
@@ -36,21 +34,25 @@ def process_image(image):
     return processed_images
 
 # Define Shiny app
-def main():
-    st.title("Sidewalk Segmentation App")
-    uploaded_image = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
+app = hf.start()
+
+@app.streamlit_app(
+    title="Sidewalk Segmentation App",
+    uploaders={"Upload an image": ["jpg", "jpeg", "png"]}
+)
+def main(uploaded_image):
+    # Convert uploaded image to PIL Image
+    image = Image.open(io.BytesIO(uploaded_image.read()))
 
-    if uploaded_image is not None:
-        # Display uploaded image
-        image = Image.open(uploaded_image)
-        st.image(image, caption="Uploaded Image", use_column_width=True)
+    # Display uploaded image
+    app.image(image, caption="Uploaded Image")
 
-        # Process uploaded image
-        with st.spinner("Processing..."):
-            segmentation_result = process_image(image)
+    # Process uploaded image
+    with app.spinner("Processing..."):
+        segmentation_result = process_image(image)
 
-        # Display segmentation result
-        st.image(segmentation_result, caption="Sidewalk Segmentation Result", use_column_width=True)
+    # Display segmentation result
+    app.image(segmentation_result, caption="Sidewalk Segmentation Result")
 
 if __name__ == "__main__":
-    main()
+    app.run()