Spaces:
Build error
Build error
Create Dockerfile
Browse files- Dockerfile +29 -0
Dockerfile
ADDED
@@ -0,0 +1,29 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# ベースイメージ: CUDA対応のPyTorch公式イメージ(HF SpacesのT4/GPUと互換)
|
2 |
+
FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime
|
3 |
+
|
4 |
+
# 作業ディレクトリ
|
5 |
+
WORKDIR /app
|
6 |
+
|
7 |
+
# Python依存ファイルをコピー
|
8 |
+
COPY requirements.txt ./
|
9 |
+
|
10 |
+
# 依存ライブラリのインストール(Gradio、FastAPI、HuggingFace関連など)
|
11 |
+
RUN apt-get update && apt-get install -y \
|
12 |
+
git wget libgl1-mesa-glx libglib2.0-0 \
|
13 |
+
&& pip install --upgrade pip \
|
14 |
+
&& pip install --no-cache-dir -r requirements.txt
|
15 |
+
|
16 |
+
# アプリケーションのファイル一式をコピー
|
17 |
+
COPY . .
|
18 |
+
|
19 |
+
# Hugging Face CLI認証用(必要に応じて)
|
20 |
+
ENV HF_HUB_DISABLE_SYMLINKS_WARNING=1
|
21 |
+
ENV TRANSFORMERS_CACHE=/app/cache
|
22 |
+
ENV HF_HOME=/app/cache
|
23 |
+
|
24 |
+
# ポート(Spaces UI用: 7860 / FastAPI用: 8000 どちらも対応)
|
25 |
+
EXPOSE 7860
|
26 |
+
EXPOSE 8000
|
27 |
+
|
28 |
+
# 実行コマンド(Gradio + FastAPI対応のapp.py)
|
29 |
+
CMD ["python", "app.py"]
|