update: description & examples

.gitattributes

@@ -1 +1,2 @@
 *.png filter=lfs diff=lfs merge=lfs -text
+*.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -24,7 +24,7 @@ description = """
     <a href="https://github.com/MapItAnywhere/MapItAnywhere" target="_blank">Code</a>
 </h3>
 <p align="center">
-Mapper generates birds-eye-view maps from first person view monocular images. Try our demo by uploading your own images.
+Mapper generates birds-eye-view maps from in-the-wild monocular first-person view images. You can try our demo by uploading your images or using the examples provided. Tip: You can also try out images across the world using <a href="https://www.mapillary.com/app" target="_blank">Mapillary</a> &#128521;
 </p>
 """
 
@@ -110,21 +110,22 @@ def run(input_img):
     gravity, camera = calibrator.run(image, exif=exif)
 
     data = preprocess_pipeline(image, gravity, camera)
-    
     res = model(data)
     
-    plot_images([image], pad=0., adaptive=True)
-    fig1 = plt.gcf()
-
     prediction = res['output']
     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
 
-    plot_images([rgb_prediction], pad=0., adaptive=True)
-    fig2 = plt.gcf()
+    plot_images([image, rgb_prediction], titles=["Input Image", "Prediction"], pad=2, adaptive=True)
+
+    return plt.gcf()
+
 
-    return fig1, fig2
+examples = [
+    ["examples/219538346289766_undistorted.jpg"],
+    ["examples/812412413010936_undistorted.jpg"],
+]
 
 demo = gr.Interface(
     fn=run,
@@ -132,8 +133,8 @@ demo = gr.Interface(
         gr.File(file_types=["image"], label="Input Image")
     ], 
     outputs=[
-        gr.Plot(label="Inputs", format="png"),
-        gr.Plot(label="Outputs", format="png"),
+        gr.Plot(label="Prediction", format="png"),
     ],
-    description=description,)
+    description=description,
+    examples=examples)
 demo.launch(share=False, server_name="0.0.0.0")