Dockerfile

FROM python:3.9

WORKDIR /app

RUN apt update && apt install sudo -y
# 必要なライブラリをインストール
RUN pip install --no-cache-dir gradio transformers torch

# 環境変数を設定（キャッシュディレクトリを変更）
ENV TRANSFORMERS_CACHE=/app/cache
# 環境変数を変更
ENV HF_HOME=/app/cache

RUN mkdir flagged && chmod 777 flagged


# モデルをクローン
RUN git clone https://huggingface.co/soiz1/animeo /app/model
#https://huggingface.co/soiz1/shirakami-hubuki-hololive

# スクリプトをコピー
COPY app.py /app/app.py

# アプリを起動
CMD ["python", "/app/app.py"]