Spaces:
Sleeping
Sleeping
Create Dockerfile
Browse files- 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"]
|