Upload app.py

app.py

@@ -102,10 +102,10 @@ with gr.Blocks(title="FireRedASR") as demo:
 
 if __name__ == "__main__":
     # Download model
-    local_dir='pretrained_models/FireRedASR2-AED-L'
-    snapshot_download(repo_id='FireRedTeam/FireRedASR2-AED-L', local_dir=local_dir)
-    local_dir_llm='pretrained_models/FireRedASR2-LLM-L'
-    snapshot_download(repo_id='FireRedTeam/FireRedASR2-LLM-L', local_dir=local_dir_llm)
+    local_dir='pretrained_models/FireRedASR2-AED'
+    snapshot_download(repo_id='FireRedTeam/FireRedASR2-AED', local_dir=local_dir)
+    local_dir_llm='pretrained_models/FireRedASR2-LLM'
+    snapshot_download(repo_id='FireRedTeam/FireRedASR2-LLM', local_dir=local_dir_llm)
     # Init model
     init_model(local_dir, local_dir_llm)
     # UI