Update models/model_loader.py

models/model_loader.py

@@ -1,12 +1,15 @@
 import os
 import streamlit as st
 
-os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
-os.environ["HF_HOME"] = "/tmp/huggingface"
+CACHE_DIR = "/tmp/pretrained_models"
+
 
 from speechbrain.pretrained.interfaces import foreign_class
 from faster_whisper import WhisperModel
 
+# Ensure the folder exists and is writable
+os.makedirs(CACHE_DIR, exist_ok=True)
+
 # -------------------------------
 # Load Model (Cached)
 # -------------------------------
@@ -20,7 +23,8 @@ def load_accent_model():
         return foreign_class(
             source="Jzuluaga/accent-id-commonaccent_xlsr-en-english",
             pymodule_file="custom_interface.py",
-            classname="CustomEncoderWav2vec2Classifier"
+            classname="CustomEncoderWav2vec2Classifier",
+            cache_dir = CACHE_DIR
         )
     except Exception as e:
         st.error(f"❌ Error loading model: {e}")