revi13 commited on
Commit
d3d8e85
·
verified ·
1 Parent(s): fd6d685

Create Dockerfile

Browse files
Files changed (1) hide show
  1. 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"]