Upload Dockerfile with huggingface_hub

Dockerfile

@@ -11,8 +11,8 @@ ENV PATH="/home/user/.local/bin:$PATH"
 
 WORKDIR /app
 
-# Create necessary directories
-RUN mkdir -p /app/heart/models /app/models
+# Create cache directory
+RUN mkdir -p /home/user/.cache/huggingface/hub
 
 # Copy requirements first to leverage Docker cache
 COPY --chown=user requirements.txt requirements.txt
@@ -24,6 +24,7 @@ COPY --chown=user . /app
 # Set environment variables
 ENV FLASK_APP=app.py
 ENV FLASK_ENV=production
+ENV HF_HOME=/home/user/.cache/huggingface
 
 # Expose the port the app runs on
 EXPOSE 7860