Update app.py

app.py

@@ -97,11 +97,17 @@ def sepia(input_img):
     pred_img = pred_img.astype(np.uint8)
 
     fig = draw_plot(pred_img, seg)
+
+    class_probabilities = {
+        labels_list[i]: probabilities[:, :, i].numpy() for i in range(len(labels_list))
+    }
+    print(class_probabilities)
+    
     return fig
 
 demo = gr.Interface(fn=sepia,
                     inputs=gr.Image(shape=(400, 600)),
-                    outputs=['plot'],
+                    outputs=['plot', "text"],
                     examples=["citiscapes-1.jpeg", "citiscapes-2.jpeg"],
                     allow_flagging='never')