Update app.py

app.py

@@ -32,5 +32,10 @@ if uploaded_file is not None:
     st.image(image, caption='Uploaded Image.', use_column_width=True)
     st.write("")
 
-    results = semantic_segmentation(image)
-    st.json(results)
+    segmentation_results = semantic_segmentation(image)
+    st.json(segmentation_results)
+
+    image_with_masks = draw_masks_fromDict(image_np, segmentation_results)
+
+    image_with_masks_pil = Image.fromarray(image_with_masks, 'RGB')
+    st.image(image_with_masks_pil, caption='Image with Masks.', use_column_width=True)