modify app file

app.py

@@ -68,11 +68,8 @@ def segment_image(input_image):
     output_image = cv2.imread('output_figure.png')
     return Image.fromarray(output_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 = Image.open('sample.jpg')
@@ -82,7 +79,7 @@ with col1:
 
 with col2:
     output = segment_image(input)
-    output = st.image()
+    st.image(output)
 
 if file is not None:
     input = file