Nutnell commited on
Commit
d6130dc
·
verified ·
1 Parent(s): 2e15747

Create Dockerfile.

Browse files
Files changed (1) hide show
  1. Dockerfile +14 -0
Dockerfile ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # 1. Use a base image with Python and CUDA (for the GPU)
2
+ FROM python:3.11-slim
3
+
4
+ # 2. Set the working directory
5
+ WORKDIR /code
6
+
7
+ # 3. Copy all your application files into the container
8
+ COPY . .
9
+
10
+ # 4. Install the dependencies
11
+ RUN pip install --no-cache-dir -r requirements.txt
12
+
13
+ # 5. Run the fine-tuning script when the Space starts
14
+ CMD ["python", "fine_tune.py"]