slav7 commited on
Commit
b06ca90
·
verified ·
1 Parent(s): 474695a

Upload 2 files

Browse files
Files changed (2) hide show
  1. src/d.py +5 -0
  2. src/streamlit_app.py +7 -1
src/d.py CHANGED
@@ -46,6 +46,11 @@ def debug_audio(audio_path):
46
  def get_accent_classifier():
47
  from speechbrain.pretrained.interfaces import foreign_class
48
  import os
 
 
 
 
 
49
  # Set all relevant cache dirs to /tmp to avoid permission errors
50
  os.environ["HF_HOME"] = "/tmp/huggingface"
51
  os.environ["HF_HUB_CACHE"] = "/tmp/huggingface/hub"
 
46
  def get_accent_classifier():
47
  from speechbrain.pretrained.interfaces import foreign_class
48
  import os
49
+ import shutil
50
+ # Remove any local folders that may cause permission errors
51
+ for folder in ["pretrained_models", "wav2vec2_checkpoints"]:
52
+ if os.path.exists(folder):
53
+ shutil.rmtree(folder)
54
  # Set all relevant cache dirs to /tmp to avoid permission errors
55
  os.environ["HF_HOME"] = "/tmp/huggingface"
56
  os.environ["HF_HUB_CACHE"] = "/tmp/huggingface/hub"
src/streamlit_app.py CHANGED
@@ -1,5 +1,11 @@
1
- import streamlit as st
2
  import os
 
 
 
 
 
 
 
3
  from d import download_and_extract_audio, analyze_accent, get_accent_classifier
4
 
5
  st.title("English Accent Classifier")
 
 
1
  import os
2
+ os.environ["HF_HOME"] = "/tmp/huggingface"
3
+ os.environ["HF_HUB_CACHE"] = "/tmp/huggingface/hub"
4
+ os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
5
+ os.environ["TORCH_HOME"] = "/tmp/torch"
6
+ os.environ["XDG_CACHE_HOME"] = "/tmp/xdg_cache"
7
+
8
+ import streamlit as st
9
  from d import download_and_extract_audio, analyze_accent, get_accent_classifier
10
 
11
  st.title("English Accent Classifier")