Sync with https://github.com/mozilla-ai/osm-ai-helper

app.py

@@ -51,14 +51,14 @@ def inference(lat_lon, margin):
             repo_type="model",
             local_dir="models",
         )
-    with st.spinner("Running inference..."):
+    with st.spinner("Downloading image and Running inference..."):
         output_path, existing, new, missed = run_inference(
             yolo_model_file="models/model.pt",
             output_dir="/tmp/results",
             lat_lon=lat_lon,
             margin=margin,
+            save_full_images=False,
         )
-        st.text(f"Found: {len(new)} new polygons")
     return output_path, existing, new
 
 
@@ -130,8 +130,15 @@ and [Finetune Model](https://colab.research.google.com/github/mozilla-ai//osm-ai
 
 st.divider()
 
-st.subheader(
-    "Click on the map to select a latitude and longitude. The model will try to find swimming pools around this location."
+st.subheader("Click on the map to select a latitude and longitude.")
+
+st.markdown(
+    """
+The model will try to find swimming pools around this location.
+
+Note that this model was trained with data from [Galicia](https://nominatim.openstreetmap.org/ui/details.html?osmtype=R&osmid=349036&class=boundary),
+so it might fail to generalize to significantly different places.
+"""
 )
 
 show_map()