FROM corex:4.3.0

WORKDIR /workspace
RUN pip install diffusers==0.34.0 sentencepiece transformers==4.55.2
COPY main.py dataset.json /workspace/