Update src/streamlit_app.py

src/streamlit_app.py

@@ -11,11 +11,15 @@ import yt_dlp
 from pydub import AudioSegment
 from pydub.silence import detect_nonsilent
 
+model_dir = "/tmp/pretrained_models"
+os.makedirs(model_dir, exist_ok=True)
+
 # Load model once
 classifier = foreign_class(
     source="Jzuluaga/accent-id-commonaccent_xlsr-en-english",
     pymodule_file="custom_interface.py",
-    classname="CustomEncoderWav2vec2Classifier"
+    classname="CustomEncoderWav2vec2Classifier",
+    savedir=model_dir
 )
 
 def extract_loom_id(url):