Update app.py

app.py

@@ -97,17 +97,12 @@ 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', "text"],
+                    outputs=['plot'],
                     examples=["citiscapes-1.jpeg", "citiscapes-2.jpeg"],
                     allow_flagging='never')