chaning to blocks

app.py

@@ -130,8 +130,9 @@ def create_integrated_gradio_interface(analyzer: MultiModelAnalyzer):
     
     model_names = analyzer.get_model_names()
     
-    with gr.Blocks(title="Discrete Token Analysis") as demo:
-        gr.Markdown("#Discrete Token Phoneme Analysis")
+    with gr.Column() as demo:
+
+        gr.Markdown("Discrete Token Phoneme Analysis")
         # gr.Markdown("Compare HuBERT models and analyze discrete representations")
         
         with gr.Tabs():
@@ -156,10 +157,10 @@ def create_integrated_gradio_interface(analyzer: MultiModelAnalyzer):
                 )
                 
                 # Initialize
-                demo.load(
-                    fn=update_comparison,
-                    outputs=[comparison_plot, metrics_table]
-                )
+                # refresh_comparison_btn.click(
+                #     fn=update_comparison,
+                #     outputs=[comparison_plot, metrics_table]
+                # )
             
             # Tab 2: Single Model Analysis
             """
@@ -1015,7 +1016,7 @@ def create_integrated_gradio_interface(analyzer: MultiModelAnalyzer):
                 #     outputs=[comparison_plot, comparison_plot]
                 # )
     
-    return demo
+    # return demo
 
 
 def create_root_interface(output_dir):