Added examples

app.py

@@ -7,8 +7,6 @@ from pathlib import Path
 from loguru import logger
 import cv2
 import torch
-import ultralytics
-from ultralytics import YOLO
 import time
 import base64
 import requests
@@ -95,7 +93,9 @@ example_data = [
     #["./demo/airport01.jpg", "aircraft", 0.25, 0.25],
     #["./demo/SPOT_Storage.jpg", "storage", 0.25, 0.25],
     #["./demo/Satellite_Image_Marina_New_Zealand.jpg", "ship", 0.25, 0.25],
-    #["./demo/Pleiades_HD15_Miami_Marina.jpg", "ship", 0.25, 0.25],
+    ["./demo/Pleiades_HD15_Miami_Marina.jpg", "motorboat", 0.3, 0.0],
+    ["./demo/Pleiades_HD15_Miami_Marina.jpg", "palm tree", 0.15, 0.3],
+    ["./demo/Pleiades_HD15_Miami_Marina.jpg", "building", 0.3, 0.0],
 ]
 
 # Define CSS for some elements
@@ -133,7 +133,7 @@ with demo:
         inputs = [input_image, text_prompt, box_threshold, text_threshold],
         outputs = [output_image, dimensions, detections, stopwatch],
         fn=predict_image,
-        cache_examples=False,
+        cache_examples=True,
         label='Try these images!'
     )
 
@@ -141,7 +141,8 @@ with demo:
                 and <a href='https://www.dl4eo.com/'>DL4EO</a>. The demonstration images are Pléiades \
                 images provided by CNES with distribution by Airbus DS. The model architecture and weights \
                 are provided <a href='https://github.com/IDEA-Research/GroundingDINO'>Grounding DINO</a>. \
-                This is a demonstration only. Please contact <a href='mailto:jeff@dl4eo.com'>me</a> \
+                The model has not been trained specifically on satellite imagery and should be finetuned for this task. \
+                This is for demonstration only. Please contact <a href='mailto:[email protected]'>me</a> \
                 for more information on how you could get access to a commercial model or API. </p>")
 
 demo.launch(