[Add, Fix] launch app, cache models

app.py

@@ -1,11 +1,13 @@
 import gradio as gr
 
-from src.dataloading import get_leaderboard_models, get_leaderboard_datasets
+
+
+from src.dataloading import get_leaderboard_models_cached, get_leaderboard_datasets
 
 
 def create_demo():
     # Fetch data once on startup (cache this in production)
-    models = get_leaderboard_models()
+    models = get_leaderboard_models_cached()
     datasets = get_leaderboard_datasets()
 
     with gr.Blocks(title="LLM Similarity Analyzer") as demo:
@@ -44,4 +46,23 @@ def create_demo():
             outputs=similarity_output
         )
         
-    return demo
+    return demo
+
+
+def create_demo_with_refresh():
+    demo = create_demo()
+    
+    with demo:
+        refresh_btn = gr.Button("Refresh Model List")
+        def refresh_models():
+            return gr.Dropdown(choices=get_leaderboard_models_cached())
+        
+        refresh_btn.click(
+            fn=refresh_models,
+            outputs=model_dropdown
+        )
+    
+    return demo
+
+demo = create_demo_with_refresh()
+demo.launch()

src/dataloading.py

@@ -1,4 +1,6 @@
 from huggingface_hub import HfApi
+from functools import lru_cache
+
 
 def get_leaderboard_models():
     api = HfApi()
@@ -20,6 +22,12 @@ def get_leaderboard_models():
     
     return sorted(list(set(models)))  # Remove duplicates
 
+
+@lru_cache(maxsize=1)
+def get_leaderboard_models_cached():
+    return get_leaderboard_models()
+
+
 def get_leaderboard_datasets():
     return [
         "ai2_arc",