added more examples

app.py

@@ -116,6 +116,8 @@ def run(input_img):
     rgb_prediction = one_hot_argmax_to_rgb(prediction, 6).squeeze(0).permute(1, 2, 0).cpu().long().numpy()
     valid = res['valid_bev'].squeeze(0)[..., :-1]
     rgb_prediction[~valid.cpu().numpy()] = 255
+    
+     # TODO: add legend here
 
     plot_images([image, rgb_prediction], titles=["Input Image", "Prediction"], pad=2, adaptive=True)
 
@@ -123,8 +125,11 @@ def run(input_img):
 
 
 examples = [
-    ["examples/219538346289766_undistorted.jpg"],
-    ["examples/812412413010936_undistorted.jpg"],
+    ["examples/left_crossing.jpg"],
+    ["examples/crossing.jpg"]
+    ["examples/two_roads.jpg"],
+    ["examples/night_road.jpg"],
+    ["examples/night_crossing.jpg"],
 ]
 
 demo = gr.Interface(