add results columns in app.py

app.py

@@ -92,12 +92,17 @@ if file_imagen is not None:
 
     output = segmentation(img)
     if output is not None:
-        st.subheader("Segmentacion: ")
-        # st.write(output.shape)
-        st.image(output[0], width=400)
-        st.subheader("Mask: ")
-        # st.write(output.shape)
-        st.image(output[1], width=400)
+        st.subheader("Original: ")
+        st.image(img, width=850)
+        col1, col2 = st.columns(2)
+
+        with col1:
+            st.subheader("Segmentacion: ")
+            st.image(output[0])
+
+        with col2:
+            st.subheader("Mask: ")
+            st.image(output[1])
 
 # article = "<div style='text-align: center;'><a href='https://keras.io/examples/vision/deeplabv3_plus/' target='_blank'>Keras example by Praveen Kaushik</a></div>"
 # gr.Interface(segmentation, i, o, examples=examples, allow_flagging=False, analytics_enabled=False,