Update app.py

app.py

@@ -14,6 +14,15 @@ import os
 import logging
 import warnings
 
+import numpy as np
+from src.retrieval.query import _compute_umap
+
+dummy = [np.random.rand(384) for _ in range(6)]
+dummy = [e / np.linalg.norm(e) for e in dummy]
+_compute_umap(dummy)
+print("UMAP warmed up")
+
+
 # ---------------- ENV + LOGGING ----------------
 os.environ["TRANSFORMERS_VERBOSITY"] = "error"
 os.environ["HF_HUB_DISABLE_PROGRESS_BARS"] = "1"