fix: image display

.gitignore

@@ -0,0 +1,6 @@
+# env files
+.env 
+.venv
+
+# cache
+*pycache*

app.py

@@ -6,7 +6,6 @@ import streamlit as st
 import torch
 import torch.nn.functional as F
 from briarmbg import BriaRMBG
-from gradio_imageslider import ImageSlider
 from PIL import Image
 from torchvision.transforms.functional import normalize
 
@@ -20,7 +19,7 @@ def resize_image(image):
 
 def process(image):
     # prepare input
-    orig_image = Image.fromarray(image)
+    orig_image = Image.open(image)
     w, h = orig_image.size
     image = resize_image(orig_image)
     im_np = np.array(image)
@@ -58,28 +57,35 @@ def main():
         unsafe_allow_html=True,
     )
 
+    # sidebar
+    with st.sidebar:
+        img_file = st.file_uploader(
+            label="Upload image",
+            type=["jpg", "png", "jpeg"],
+            key="image_file_uploader",
+        )
+
     cols = st.columns(2)
 
     with cols[0]:
-        with st.container(border=True, height=300):
-            img_file = st.file_uploader(
-                label="Upload image",
-                type=["jpg", "png", "jpeg"],
-                key="image_file_uploader",
-            )
+        with st.container(border=True, height=600):
             if img_file:
-                st.Image(img_file)
+                st.image(img_file)
+            else:
+                st.info("Drag and drop the sample image into upload sidebar", icon="💡")
         sub_btn = st.button("Remove bg", key="sub_btn")
+
     with cols[1]:
-        with st.container(border=True, height=400):
+        with st.container(border=True, height=600):
             if sub_btn and img_file:
-                processed_img = process(img_file.read())
-                st.Image(processed_img)
+                processed_img = process(img_file)
+                st.image(processed_img)
             else:
                 st.write("Waiting for image...")
 
-    with st.container(border=True, height=200):
-        st.Image("input.jpg")
+    with st.container(border=True, height=400):
+        st.write("Sample image")
+        st.image("input.jpg")
 
 
 if __name__ == "__main__":

requirements.txt

@@ -1,5 +1,3 @@
-gradio
-gradio_imageslider
 torch
 torchvision
 pillow