DragonProgrammer commited on
Commit
4981ad2
·
verified ·
1 Parent(s): 864174a

Create Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +14 -0
Dockerfile ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Use Hugging Face’s prebuilt PyTorch + Transformers + CUDA image
2
+ FROM huggingface/transformers-pytorch-gpu:latest
3
+
4
+ # Copy requirements and install minimal dependencies
5
+ COPY requirements.txt /app/requirements.txt
6
+ RUN pip install --no-cache-dir -r /app/requirements.txt && \
7
+ apt-get clean && rm -rf /var/lib/apt/lists/*
8
+
9
+ # Copy your app files
10
+ COPY . /app
11
+ WORKDIR /app
12
+
13
+ # Default command
14
+ CMD ["python3", "app.py"]