Update app.py

app.py

@@ -78,6 +78,7 @@ def run_timeline(repo_url: str, threshold: float, multimodal: bool, sim_method:
 CUSTOM_CSS = """
 #graph_html iframe, #timeline_html iframe {height:85vh !important; width:100% !important; border:none;}
 """
+TAB_INDEX = {"timeline": 0, "loc": 1, "graph": 2}
 
 with gr.Blocks(css=CUSTOM_CSS) as demo:
     gr.Markdown("## 🔍 Modular-candidate explorer for 🤗 Transformers")
@@ -126,10 +127,12 @@ with gr.Blocks(css=CUSTOM_CSS) as demo:
             )
 
     def _select_tab_on_load(req: gr.Request):
-        tab = (req.query_params or {}).get("tab")
-        if tab in {"timeline", "loc", "graph"}:
-            return gr.update(selected=tab)
-        return gr.update()
+        qp = req.query_params or {}
+        key = (qp.get("tab") or "").lower()
+        idx = TAB_INDEX.get(key)
+        if idx is not None:
+            return gr.Tabs.update(selected=idx)
+        return gr.Tabs.update()
 
     demo.load(_select_tab_on_load, outputs=tabs)