fix bug

.gitignore

@@ -1,16 +1,14 @@
-./ConST
-./fairseq
-./fairseq_cli
-./fairseq.egg_info
+./ConST/
+./fairseq/
+./fairseq_cli/
+./fairseq.egg_info/
 hubconf.py
 LICENSE
 pyproject.toml
-./build
 setup.py
 train.py
 requirements.apt.txt
-./ConST
 ./.eggs
 ./.idea
-./build
-./docs
+./build/
+./docs/

app.py

@@ -12,7 +12,7 @@ import shutil
 import yaml
 import torchaudio
 import gradio as gr
-from huggingface_hub import snapshot_download
+from huggingface_hub import snapshot_download, hf_hub_url
 
 
 LANGUAGE_CODES = {
@@ -40,8 +40,8 @@ LANG_GEN_SETUPS = {
 os.system("git clone https://github.com/ReneeYe/ConST")
 os.system('mv ConST/* ./')
 os.system("rm -rf ConST")
-os.system("python3 setup.py install")
-os.system("python3 setup.py build_ext --inplace")
+os.system("sudo python3 setup.py install")
+os.system("sudo python3 setup.py build_ext --inplace")
 os.system("mkdir -p data checkpoint")
 
 
@@ -71,7 +71,7 @@ def get_vocab_and_yaml(language):
 
     # write yaml file
     abs_path = os.popen("pwd").read().strip()
-    yaml_dict = LANG_GEN_SETUPS["tgt_lang"]
+    yaml_dict = LANG_GEN_SETUPS[tgt_lang]
     yaml_dict["input_channels"] = 1
     yaml_dict["use_audio_input"] = True
     yaml_dict["prepend_tgt_lang_tag"] = True