Update Dockerfile

Dockerfile

@@ -16,6 +16,12 @@ RUN apt-get update && apt-get install -y \
 RUN pip install --no-cache-dir --upgrade pip
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Set environment variable for Hugging Face cache
+ENV HF_HOME=/app/cache
+
+# Create the cache directory
+RUN mkdir -p /app/cache
+
 # Expose port 8501 for Streamlit and port 5000 for Flask
 EXPOSE 8501
 EXPOSE 5000