moarafa97 commited on
Commit
eeb4134
·
verified ·
1 Parent(s): 4f5bf2c

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -2
src/streamlit_app.py CHANGED
@@ -10,11 +10,10 @@ import torchaudio
10
  from speechbrain.pretrained.interfaces import foreign_class
11
 
12
 
13
- # Load model using custom interface
14
  @st.cache_resource
15
  def load_model():
16
  try:
17
- os.environ["SPEECHBRAIN_CACHE"] = os.path.join(os.getcwd(), "models")
18
  return foreign_class(
19
  source="Jzuluaga/accent-id-commonaccent_xlsr-en-english",
20
  pymodule_file="custom_interface.py",
 
10
  from speechbrain.pretrained.interfaces import foreign_class
11
 
12
 
 
13
  @st.cache_resource
14
  def load_model():
15
  try:
16
+ os.environ["SPEECHBRAIN_CACHE"] = "/data/speechbrain"
17
  return foreign_class(
18
  source="Jzuluaga/accent-id-commonaccent_xlsr-en-english",
19
  pymodule_file="custom_interface.py",