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

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +2 -1
src/streamlit_app.py CHANGED
@@ -17,7 +17,8 @@ def load_model():
17
  return foreign_class(
18
  source="Jzuluaga/accent-id-commonaccent_xlsr-en-english",
19
  pymodule_file="custom_interface.py",
20
- classname="CustomEncoderWav2vec2Classifier"
 
21
  )
22
  except Exception as e:
23
  st.error(f"❌ Model failed to load: {e}")
 
17
  return foreign_class(
18
  source="Jzuluaga/accent-id-commonaccent_xlsr-en-english",
19
  pymodule_file="custom_interface.py",
20
+ classname="CustomEncoderWav2vec2Classifier",
21
+ savedir="/data/speechbrain/accent-id"
22
  )
23
  except Exception as e:
24
  st.error(f"❌ Model failed to load: {e}")