Update app.py

app.py

@@ -126,31 +126,31 @@ with gr.Blocks(css=CUSTOM_CSS) as demo:
             )
         embed_html = gr.HTML(visible=False) 
 
-def _on_load(req: gr.Request):
-    qp = req.query_params or {}
-    tab_key = (qp.get("tab") or "").lower()
-    embed = (qp.get("embed") == "1")
-
-    tab_sel = TAB_INDEX.get(tab_key, 0)
-
-    if embed:
-        if tab_key == "graph":
-            html, _ = run_graph(HF_MAIN_REPO, 0.7, False, "jaccard")
-        elif tab_key == "timeline":
-            html, _ = run_timeline(HF_MAIN_REPO, 0.7, False, "jaccard")
-        else:  # "loc" or anything else
-            html = run_loc("jaccard", False)
-
+    def _on_load(req: gr.Request):
+        qp = req.query_params or {}
+        tab_key = (qp.get("tab") or "").lower()
+        embed = (qp.get("embed") == "1")
+    
+        tab_sel = TAB_INDEX.get(tab_key, 0)
+    
+        if embed:
+            if tab_key == "graph":
+                html, _ = run_graph(HF_MAIN_REPO, 0.7, False, "jaccard")
+            elif tab_key == "timeline":
+                html, _ = run_timeline(HF_MAIN_REPO, 0.7, False, "jaccard")
+            else:  # "loc" or anything else
+                html = run_loc("jaccard", False)
+    
+            return (
+                gr.update(visible=False),        # header
+                gr.update(visible=False),        # tabs
+                gr.update(value=html, visible=True),  # embed_html
+            )
+    
         return (
-            gr.update(visible=False),        # header
-            gr.update(visible=False),        # tabs
-            gr.update(value=html, visible=True),  # embed_html
+            gr.update(visible=True),           # header
+            gr.update(visible=True, selected=tab_sel),  # tabs
+            gr.update(visible=False),          # embed_html
         )
-
-    return (
-        gr.update(visible=True),           # header
-        gr.update(visible=True, selected=tab_sel),  # tabs
-        gr.update(visible=False),          # embed_html
-    )
-
-demo.load(_on_load, outputs=[header, tabs, embed_html])
+    
+    demo.load(_on_load, outputs=[header, tabs, embed_html])