modify app file

app.py

@@ -71,20 +71,24 @@ def segment_image(input_image):
 ## Manipularemos la interfaz para que podamos usar imágenes ejemplo
 ## Si el usuario da click en un ejemplo entonces el modelo correrá con él
 images = ["sample.jpg"]
-
+file = st.file_uploader("Upload File", type=["png", "jpg", "jpeg"])
 ## Creamos tres columnas; en cada una estará una imagen ejemplo
 col1, col2 = st.columns(2)
 with col1:
-    input = st.image()
+    input = Image.open('sample.jpg')
+    st.image(input, caption='Test Image provided by SAM github')     
     # ex = Image.open(images[0])
     # st.image(ex, width=200)
 
 with col2:
+    output = segment_image(input)
     output = st.image()
 
-if st.button("Generate Mask"):
-    output = segment_image(input)
-    if output is not None:
-        st.subheader("Segmentation Result")
-        st.write(output.shape)
-        st.image(output, width=850)
+if file is not None:
+    input = file
+    if st.button("Generate Mask"):
+        output = segment_image(input)
+        if output is not None:
+            st.subheader("Segmentation Result")
+            st.write(output.shape)
+            st.image(output, width=850)