fix

app.py

@@ -74,71 +74,69 @@ with demo:
             )
             gr.HTML(pretrained_fig.data)
 
-            with gr.TabItem(label="Instructions Finetuend Models"):
-                inst_finetuned_fig, ax = plt.subplots(figsize=(12, 6))
-                ax.set_xlim(0, 100)
-                inst_finetuned_dataset = open_llm_race_dataset[
-                    open_llm_race_dataset["type"] == ModelType.IFT.name
-                ]
-                inst_finetuned_dataset = inst_finetuned_dataset.pivot(
-                    index="date", columns="model", values="score"
-                )
-                inst_finetuned_dataset.fillna(0, inplace=True)
-                inst_finetuned_fig = bcr.bar_chart_race(
-                    inst_finetuned_dataset,
-                    n_bars=10,
-                    fixed_max=True,
-                    period_length=1000,
-                    steps_per_period=20,
-                    end_period_pause=100,
-                    bar_texttemplate="{x:.2f}",
-                    filter_column_colors=True,
-                    fig=inst_finetuned_fig,
-                )
-                gr.HTML(inst_finetuned_fig.data)
+        with gr.TabItem(label="Instructions Finetuend Models"):
+            inst_finetuned_fig, ax = plt.subplots(figsize=(12, 6))
+            ax.set_xlim(0, 100)
+            inst_finetuned_dataset = open_llm_race_dataset[
+                open_llm_race_dataset["type"] == ModelType.IFT.name
+            ]
+            inst_finetuned_dataset = inst_finetuned_dataset.pivot(
+                index="date", columns="model", values="score"
+            )
+            inst_finetuned_dataset.fillna(0, inplace=True)
+            inst_finetuned_fig = bcr.bar_chart_race(
+                inst_finetuned_dataset,
+                n_bars=10,
+                fixed_max=True,
+                period_length=1000,
+                steps_per_period=20,
+                end_period_pause=100,
+                bar_texttemplate="{x:.2f}",
+                filter_column_colors=True,
+                fig=inst_finetuned_fig,
+            )
+            gr.HTML(inst_finetuned_fig.data)
 
-            with gr.TabItem(label="RLHF Models"):
-                rl_fig, ax = plt.subplots(figsize=(12, 6))
-                ax.set_xlim(0, 100)
-                rl_dataset = open_llm_race_dataset[
-                    open_llm_race_dataset["type"] == ModelType.IFT.name
-                ]
-                rl_dataset = rl_dataset.pivot(
-                    index="date", columns="model", values="score"
-                )
-                rl_dataset.fillna(0, inplace=True)
-                rl_fig = bcr.bar_chart_race(
-                    rl_dataset,
-                    n_bars=10,
-                    fixed_max=True,
-                    period_length=1000,
-                    steps_per_period=20,
-                    end_period_pause=100,
-                    bar_texttemplate="{x:.2f}",
-                    filter_column_colors=True,
-                    fig=rl_fig,
-                )
-                gr.HTML(rl_fig.data)
+        with gr.TabItem(label="RLHF Models"):
+            rl_fig, ax = plt.subplots(figsize=(12, 6))
+            ax.set_xlim(0, 100)
+            rl_dataset = open_llm_race_dataset[
+                open_llm_race_dataset["type"] == ModelType.IFT.name
+            ]
+            rl_dataset = rl_dataset.pivot(index="date", columns="model", values="score")
+            rl_dataset.fillna(0, inplace=True)
+            rl_fig = bcr.bar_chart_race(
+                rl_dataset,
+                n_bars=10,
+                fixed_max=True,
+                period_length=1000,
+                steps_per_period=20,
+                end_period_pause=100,
+                bar_texttemplate="{x:.2f}",
+                filter_column_colors=True,
+                fig=rl_fig,
+            )
+            gr.HTML(rl_fig.data)
 
-            # with gr.TabItem(label="Finetuned Models"):
-            #     finetuned_dataset = open_llm_race_dataset[
-            #         open_llm_race_dataset["type"] == ModelType.FT.name
-            #     ]
-            #     finetuned_dataset = finetuned_dataset.pivot(
-            #         index="date", columns="model", values="score"
-            #     )
-            # finetuned_fig = bcr.bar_chart_race(
-            #     finetuned_dataset,
-            #     n_bars=10,
-            #     fixed_max=True,
-            #     period_length=1000,
-            #     steps_per_period=20,
-            #     end_period_pause=100,
-            #     bar_texttemplate="{x:.2f}",
-            #     filter_column_colors=True,
-            #     fig=pretrained_fig,
-            # )
-            # gr.HTML(finetuned_fig.data)
+        # with gr.TabItem(label="Finetuned Models"):
+        #     finetuned_dataset = open_llm_race_dataset[
+        #         open_llm_race_dataset["type"] == ModelType.FT.name
+        #     ]
+        #     finetuned_dataset = finetuned_dataset.pivot(
+        #         index="date", columns="model", values="score"
+        #     )
+        # finetuned_fig = bcr.bar_chart_race(
+        #     finetuned_dataset,
+        #     n_bars=10,
+        #     fixed_max=True,
+        #     period_length=1000,
+        #     steps_per_period=20,
+        #     end_period_pause=100,
+        #     bar_texttemplate="{x:.2f}",
+        #     filter_column_colors=True,
+        #     fig=pretrained_fig,
+        # )
+        # gr.HTML(finetuned_fig.data)
 
 
 def restart_space():