add model output in app.py

app.py

@@ -87,8 +87,17 @@ with col3:
     if st.button("Corre ejemplo 3"):
         file_imagen = examples[2]
 
-# if archivo_imagen is not None:
-
+if file_imagen is not None:
+    img = Image.open(file_imagen)
+
+    output = segmentation(img)
+    if output is not None:
+        st.subheader("Segmentacion: ")
+        st.write(output.shape)
+        st.image(output[0], width=850)
+        st.subheader("Mask: ")
+        st.write(output.shape)
+        st.image(output[1], width=850)
 
 # 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,