Update app.py

app.py

@@ -8,7 +8,7 @@ import torch.nn as nn
 
 EXPORT_DIR = "."
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
-TOP_K = 3
+TOP_K = 5
 
 # ---------------- Load metadata ----------------
 metadata = json.load(open(os.path.join(EXPORT_DIR, "metadata.json")))
@@ -124,11 +124,21 @@ def predict_geohash_map(img: Image.Image, true_lat=None, true_lon=None):
 
 # ---------------- Gradio UI ----------------
 with gr.Blocks() as demo:
+    gr.Markdown(
+        """
+        # 🌍 Locus - GeoGuessr Image to Coordinates Model  
+        Upload a **streetview-style image** (photo or screenshot).  
+        The model can handle images with a small UI overlay (like from Google Street View),  
+        as long as the **main scene is clearly visible**.  
+
+        Optionally, provide the real coordinates to compare predictions.
+        """
+    )
     with gr.Row():
         img_in = gr.Image(type="pil", label="Upload Streetview Image")
         with gr.Column():
-            lat_in = gr.Number(label="True Latitude (optional)")
-            lon_in = gr.Number(label="True Longitude (optional)")
+            lat_in = gr.Number(label="True Latitude (optional)", info="Enter the known latitude if available.")
+            lon_in = gr.Number(label="True Longitude (optional)", info="Enter the known longitude if available.")
     out_text = gr.Textbox(label="Predictions + Distances")
     out_plot = gr.Plot(label="Map")