Update app.py

app.py

@@ -1,9 +1,21 @@
+import os
 import subprocess
 import gradio as gr
 
-#model_file = "best_model_4800it_40M.bin"
+def list_bin_files():
+    # Get the current directory
+    current_directory = os.getcwd()
 
-model_list = ["best_model_4800it_40M.bin", "model_2200.bin", "model_4200.bin", "model_2200_tr_corpus.bin", "model_8k.bin", "model_1600.bin", "model_3600.bin"]
+    # List all files in the current directory
+    all_files = os.listdir(current_directory)
+
+    # Filter files with the ".bin" extension
+    bin_files = [file for file in all_files if file.endswith(".bin")]
+
+    return bin_files
+
+# Call the function to get the list of ".bin" files
+model_list = list_bin_files()
 
 try:
     subprocess.run(["make", "runfast"], check=True, shell=True)