example_imgs

app.py

@@ -52,10 +52,7 @@ model_path = get_asset_path(Path("models/best.onnx"))
 
 data_dir=get_asset_path(Path("img"))
 data_dir.exists()
-
-train_images = list(data_dir.glob('shells.png'))
-
-default_images = {Path(img).stem: str(img) for img in train_images}
+example_imgs = data_dir.glob("*.jpg")
 
 class PointSelector:
     def __init__(self, image=None):