Update Dockerfile

Dockerfile

@@ -3,6 +3,8 @@ FROM python:3.10-slim
 # Set environment variables
 ENV PYTHONDONTWRITEBYTECODE=1
 ENV PYTHONUNBUFFERED=1
+ENV HF_HOME=/code/.cache/huggingface
+ENV TRANSFORMERS_CACHE=/code/.cache/huggingface
 
 # Set work directory
 WORKDIR /code
@@ -18,8 +20,8 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy project files
 COPY . /code/
 
-# Expose port (Flask default is 5000)
+# Expose port
 EXPOSE 7860
 
 # Run the app
-CMD ["python", "app.py"]
+CMD ["python", "app.py"]