Update app.py

app.py

@@ -28,6 +28,8 @@ def visualize_model_output(prediction):
 
     output = np.zeros(prediction.shape)
 
+    print(output.shape,'shape of output')
+
     for unq_class in unique_classes:
         rgb_class_unique = rgb_colors[str(unq_class)]
         output[:,:,0] = rgb_class_unique[0]