mike23415 commited on
Commit
6fd7697
·
verified ·
1 Parent(s): e116825

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -5
Dockerfile CHANGED
@@ -4,14 +4,12 @@ FROM python:3.9
4
  # Set the working directory inside the container
5
  WORKDIR /app
6
 
7
- # Set environment variables to configure cache location
8
- ENV TRANSFORMERS_CACHE=/tmp/cache
9
  RUN mkdir -p /tmp/cache && chmod 777 /tmp/cache
10
 
11
  # Install system dependencies
12
- RUN apt-get update && apt-get install -y \
13
- git \
14
- && rm -rf /var/lib/apt/lists/*
15
 
16
  # Install Python dependencies
17
  COPY requirements.txt requirements.txt
 
4
  # Set the working directory inside the container
5
  WORKDIR /app
6
 
7
+ # Set environment variables for Hugging Face cache
8
+ ENV HF_HOME=/tmp/cache
9
  RUN mkdir -p /tmp/cache && chmod 777 /tmp/cache
10
 
11
  # Install system dependencies
12
+ RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
 
 
13
 
14
  # Install Python dependencies
15
  COPY requirements.txt requirements.txt