add model output in app.py

app.py

@@ -94,10 +94,10 @@ if file_imagen is not None:
     if output is not None:
         st.subheader("Segmentacion: ")
         # st.write(output.shape)
-        st.image(output[0], width=850)
+        st.image(output[0], width=400)
         st.subheader("Mask: ")
         # st.write(output.shape)
-        st.image(output[1], width=850)
+        st.image(output[1], width=400)
 
 # 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,