Update app.py

app.py

@@ -35,7 +35,7 @@ if uploaded_file is not None:
     segmentation_results = semantic_segmentation(image)
     st.json(segmentation_results)
 
-    image_with_masks = draw_masks_fromDict(image_np, segmentation_results)
+    image_with_masks = draw_masks_fromDict(np.array(image), 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)