Princess3 commited on
Commit
c46eb0d
·
verified ·
1 Parent(s): 628b309

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +8 -5
Dockerfile CHANGED
@@ -17,20 +17,23 @@ RUN apt-get update && apt-get install -y \
17
  # Create a working directory
18
  WORKDIR /app
19
 
20
- # Create cache directory
21
- RUN mkdir -p /cache && chmod +x /cache
22
 
23
  # Set the cache environment variable
24
- ENV TRANSFORMERS_CACHE=/cache
 
 
 
25
 
26
  # Copy the current directory contents into the container at /app
27
  COPY . /app
28
 
29
  # Install requirements
30
- RUN pip install --upgrade pip && pip install -r /app/requirements.txt
31
 
32
  # Unzip the data.zip file into the /app/data directory
33
  RUN unzip data.zip -d /app/data
34
 
35
  # Set the default command to run when starting the container
36
- CMD ["python", "2.py"]
 
17
  # Create a working directory
18
  WORKDIR /app
19
 
20
+ # Create cache directory with proper permissions
21
+ RUN mkdir -p /app/cache && chmod -R 777 /app/cache
22
 
23
  # Set the cache environment variable
24
+ ENV HF_HOME=/app/cache
25
+
26
+ # Install Python dependencies
27
+ RUN pip install --upgrade pip
28
 
29
  # Copy the current directory contents into the container at /app
30
  COPY . /app
31
 
32
  # Install requirements
33
+ RUN pip install -r requirements.txt
34
 
35
  # Unzip the data.zip file into the /app/data directory
36
  RUN unzip data.zip -d /app/data
37
 
38
  # Set the default command to run when starting the container
39
+ CMD ["python", "m5.py"]