Update app.py

app.py

@@ -1,7 +1,7 @@
 import pandas as pd
 import gradio as gr
 
-def compare_csv_files():
+def compare_csv_files(max_num):
     df1 = pd.read_csv("fish-speech-1.5.csv")
     df2 = pd.read_csv("fish-speech-1.4.csv")
     
@@ -11,21 +11,21 @@ def compare_csv_files():
     merged_df["CharacterErrorRate_Diff"] = merged_df["CharacterErrorRate_1.5"] - merged_df["CharacterErrorRate_1.4"]
     
     merged_df["WordErrorRate_Comparison"] = merged_df["WordErrorRate_Diff"].apply(
-        lambda x: "1.4 is the same as 1.5 (Ignored due to large diff)" if abs(x) > 10 else (
+        lambda x: "1.4 is the same as 1.5 (Ignored due to large diff)" if abs(x) > max_num else (
             f"1.5 is stronger than 1.4 ({x:.8f})" if x < 0 else (
                 f"1.4 is stronger than 1.5 ({-x:.8f})" if x > 0 else "1.4 is the same as 1.5 (0)"
             )
         )
     )
     merged_df["CharacterErrorRate_Comparison"] = merged_df["CharacterErrorRate_Diff"].apply(
-        lambda x: "1.4 is the same as 1.5 (Ignored due to large diff)" if abs(x) > 10 else (
+        lambda x: "1.4 is the same as 1.5 (Ignored due to large diff)" if abs(x) > max_num else (
             f"1.5 is stronger than 1.4 ({x:.8f})" if x < 0 else (
                 f"1.4 is stronger than 1.5 ({-x:.8f})" if x > 0 else "1.4 is the same as 1.5 (0)"
             )
         )
     )
     
-    avg_word_diff = merged_df["WordErrorRate_Diff"].loc[merged_df["WordErrorRate_Diff"].abs() <= 10].mean()
+    avg_word_diff = merged_df["WordErrorRate_Diff"].loc[merged_df["WordErrorRate_Diff"].abs() <= max_num].mean()
     avg_char_diff = merged_df["CharacterErrorRate_Diff"].loc[merged_df["CharacterErrorRate_Diff"].abs() <= 1].mean()
     overall_summary = f"""
     <h3>Overall Comparison:</h3>
@@ -42,9 +42,10 @@ def compare_csv_files():
     
     return overall_summary + result.to_html(index=False)
 
+max_num = gr.number(Number=10)
 gr.Interface(
     fn=compare_csv_files,
-    inputs=None,
+    inputs=[max_num],
     outputs="html",
     title="Fish Speech Benchmark",
     description="This is a non official model performance test from Fish Speech / Whisper Base / More data will be added later (not too much)"