cache data for map

app.py

@@ -10,7 +10,6 @@ from safetensors import safe_open
 from sentence_transformers import SentenceTransformer
 
 from semantic_search import predict
-from utils.process_data import add_coor, load_data, merge_geoemtry
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
 CITIES_ENRICHED = os.path.join("data", "cities_enriched_manually.csv")
@@ -151,10 +150,9 @@ fig = go.Figure(
 fig.update_layout(margin=dict(t=50, l=25, r=25, b=25))
 fig.update_layout(height=1000, width=1000, template="plotly")
 
+
 # load data ready to plot for local testing
-germany = pd.read_csv(MAP_PATH)
-germany["lat"] = pd.to_numeric(germany["lat"])
-germany["lon"] = pd.to_numeric(germany["lon"])
+
 # germany.drop(columns=["lat", "lon"], inplace=True)
 
 # # or generate it directly in this script
@@ -164,6 +162,34 @@ germany["lon"] = pd.to_numeric(germany["lon"])
 # germany.to_csv(MAP_PATH_WITH_COORD, index=False)
 
 # # germany need columns with lat and lon as well as hover data
+
+
+tensors = {}
+with safe_open("corpus_embeddings.pt", framework="pt", device="cpu") as f:
+    for k in f.keys():
+        tensors[k] = f.get_tensor(k)
+
+model = SentenceTransformer(
+    model_name_or_path="and-effect/musterdatenkatalog_clf",
+    device="cpu",
+    use_auth_token=HF_TOKEN,
+)
+
+
+st.set_page_config(layout="wide")
+
+
+@st.cache_data
+def load_data() -> pd.DataFrame:
+    germany = pd.read_csv(MAP_PATH)
+    return germany
+
+
+germany = load_data()
+germany["lat"] = pd.to_numeric(germany["lat"])
+germany["lon"] = pd.to_numeric(germany["lon"])
+
+
 fig_map = px.scatter_geo(
     germany,
     lat="lat",
@@ -199,20 +225,6 @@ fig_map.update_layout(
     height=700,
 )
 
-tensors = {}
-with safe_open("corpus_embeddings.pt", framework="pt", device="cpu") as f:
-    for k in f.keys():
-        tensors[k] = f.get_tensor(k)
-
-model = SentenceTransformer(
-    model_name_or_path="and-effect/musterdatenkatalog_clf",
-    device="cpu",
-    use_auth_token=HF_TOKEN,
-)
-
-
-st.set_page_config(layout="wide")
-
 st.title("Musterdatenkatalog (MDK)")
 
 st.markdown(