docker-setup

app.py

@@ -4,10 +4,16 @@ from pydub import AudioSegment  # 変換用にpydubをインポート
 import os
 import shutil
 from process import AudioProcessor
+from dotenv import load_dotenv
 
 process=AudioProcessor()
 app = Flask(__name__)
 
+load_dotenv()  # .env ファイルを読み込む
+HUGGINGFACE_HUB_TOKEN = os.getenv("HUGGINGFACE_HUB_TOKEN")
+if not HUGGINGFACE_HUB_TOKEN:
+    raise ValueError("HUGGINGFACE_HUB_TOKEN が設定されていません。")
+
 # トップページ（テンプレート: index.html）
 @app.route('/')
 @app.route('/index', methods=['GET', 'POST'])

process.py

@@ -9,7 +9,7 @@ from pyannote.audio import Model, Inference
 from pydub import AudioSegment
 class AudioProcessor():
     def __init__(self,cache_dir = "/tmp/hf_cache"):
-        hf_token = os.environ.get("HF")
+        hf_token = os.environ.get("HUGGINGFACE_HUB_TOKEN")
         if hf_token is None:
             raise ValueError("HUGGINGFACE_HUB_TOKEN が設定されていません。")
         os.makedirs(cache_dir, exist_ok=True)

requirements.txt

@@ -3,3 +3,4 @@ pyannote.audio==2.1.1
 numpy==1.23.5
 pydub==0.25.1
 matplotlib==3.6.3
+python-dotenv