update hf repo

Dockerfile

@@ -2,6 +2,8 @@ FROM python:3.9-slim
 
 WORKDIR /app
 
+
+
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
     git \
@@ -11,13 +13,12 @@ RUN apt-get update && apt-get install -y \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
 
-# Create NLTK data directory with correct permissions
+# Create and set permissions for cache directories
+RUN mkdir -p /.cache && chmod 777 /.cache
+RUN mkdir -p /root/.cache && chmod 777 /root/.cache
 RUN mkdir -p /usr/local/share/nltk_data && \
     chmod -R 777 /usr/local/share/nltk_data
 
-# Set NLTK_DATA environment variable
-ENV NLTK_DATA=/usr/local/share/nltk_data
-
 # Copy requirements file
 COPY requirements.txt .
 
@@ -49,10 +50,12 @@ RUN touch /app/versa/.installation_complete
 
 # Set port
 EXPOSE 7860
-ENV GRADIO_SERVER_NAME="0.0.0.0"
 
 # Set environment variables
+ENV GRADIO_SERVER_NAME="0.0.0.0"
+ENV NLTK_DATA=/usr/local/share/nltk_data
 ENV PYTHONUNBUFFERED=1
+ENV HF_HOME=/root/.cache/huggingface
 
 # Set for Numba cache (error from https://github.com/librosa/librosa/issues/1156)
 RUN mkdir -m 777 /tmp/NUMBA_CACHE_DIR /tmp/MPLCONFIGDIR

app.py

@@ -329,6 +329,9 @@ def evaluate_audio(gt_file, pred_file, metric_config, include_timestamps=False):
             # Pass through the NLTK_DATA environment variable
             env["NLTK_DATA"] = nltk_data_dir
 
+            # Set huggingface cache
+            env["HF_HOME"] = "/root/.cache/huggingface"
+
             process = subprocess.run(
                 cmd,
                 check=True,