File size: 677 Bytes
2f14af4
99c4e1a
dfe2ac6
18ad2c3
0dea52b
18ad2c3
efe4fae
18ad2c3
b5d949c
dfe2ac6
18ad2c3
0dea52b
dfe2ac6
 
 
 
 
 
 
 
 
 
 
 
f19377e
 
dfe2ac6
 
 
 
 
0dea52b
 
db368f0
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
FROM python:3.10-slim

# system deps
RUN apt-get update && apt-get install -y git build-essential && rm -rf /var/lib/apt/lists/*

WORKDIR /app

COPY . /app

# pip upgrade
RUN pip install --no-cache-dir --upgrade pip

# 🔥 FIXED: compatible transformers version
RUN pip install --no-cache-dir \
    torch==2.1.2+cpu \
    torchvision \
    torchaudio \
    --index-url https://download.pytorch.org/whl/cpu

RUN pip install --no-cache-dir \
    transformers==4.38.2 \
    gradio \
    accelerate \
    sentencepiece \
    huggingface_hub \
    flask

# optional (HF token support)
ENV HF_HOME=/app/cache
ENV TRANSFORMERS_CACHE=/app/cache

EXPOSE 7860

CMD ["python", "app2.py"]