Upload 2 files

Dockerfile

@@ -4,6 +4,9 @@ FROM python:3.11-slim
 # 作業ディレクトリを設定
 WORKDIR /app
 
+# .cacheディレクトリを作成
+RUN mkdir -p /.cache && chmod 777 /.cache
+
 # 必要なパッケージをインストール
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt

whisper_fastapi_online_server.py

@@ -17,6 +17,8 @@ import logging
 import logging.config
 from transformers import pipeline
 
+os.environ['HF_HOME'] = './.cache'
+
 MODEL_NAME = 'Helsinki-NLP/opus-tatoeba-en-ja'
 TRANSLATOR = pipeline('translation', model=MODEL_NAME, device='cuda')
 TRANSLATOR('Warming up!')