commit

app.py

@@ -1,8 +1,9 @@
+#credit to https://github.com/NielsRogge/Transformers-Tutorials/blob/master/SegFormer/Segformer_inference_notebook.ipynb for segmentation code
 import streamlit as st
 
 st.title("Segmentation of Beauty Products")
 
-file_name = st.file_uploader("Upload a a beauty product")
+file_name = st.file_uploader("Upload an image of a beauty product")
 
 
 from transformers import SegformerImageProcessor, SegformerForSemanticSegmentation
@@ -21,7 +22,7 @@ from PIL import Image
 
 if file_name is not None:
     image = Image.open(file_name)
-    image.show()
+    st.image(image, caption='Image without Segmentation')
 
     pixel_values = feature_extractor(image, return_tensors="pt").pixel_values.to(device)
 
@@ -89,4 +90,4 @@ if file_name is not None:
         
         plt.figure(figsize=(15, 10))
         plt.imshow(img)
-        plt.show()
+        st.plyplot(img)