File size: 399 Bytes
108dadc
2af1d99
108dadc
2af1d99
 
 
 
108dadc
 
 
 
5f695ef
2af1d99
 
2fbe8e0
2af1d99
41f0c4a
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
FROM nvcr.io/nvidia/pytorch:24.06-py3

WORKDIR /app

COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# Fix: make models and cache dirs writable
RUN mkdir -p /app/models /app/.cache && chmod -R 777 /app/models /app/.cache

ENV HF_HOME=/app/.cache

COPY . .

EXPOSE 7860

CMD python server.py --model-repo SmilingWolf/wd-eva02-large-tagger-v3 --port 7860 --token $HF_TOKEN