Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- 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}")
|