Dockerfile

# Use the base image from your Docker container
FROM trcoot/cpu-casuallm:latest

# Install any additional dependencies or files needed
# RUN pip install -r requirements.txt

# Set the environment variable for the cache directory
ENV TRANSFORMERS_CACHE=/app/.cache

# Create the cache directory and give the appropriate permissions
RUN mkdir -p /app/.cache && chmod 777 /app/.cache


# Start the application (modify this based on your app's entry point)
CMD ["python", "app.py"]