Update transcription.py

transcription.py

@@ -1,23 +1,21 @@
 import os
 from faster_whisper import WhisperModel
 from pydub import AudioSegment
+
 # Matplotlibのキャッシュディレクトリを変更
 os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
 
 # Hugging Faceのキャッシュディレクトリを変更
 os.environ["HF_HOME"] = "/tmp/huggingface"
+os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/huggingface"
 
 class TranscriptionMaker():
-    #書き起こしファイルを吐き出すディレクトリを指定
-    def __init__(self,output_dir="/tmp/data/transcriptions"):
-        self.model = WhisperModel("base", device="cpu")
+    # 書き起こしファイルを吐き出すディレクトリを指定
+    def __init__(self, output_dir="/tmp/data/transcriptions"):
+        self.model = WhisperModel("base", device="cpu", download_root="/tmp/huggingface")
         self.output_dir = output_dir
-        try:
-            if not os.path.exists(self.output_dir):
-                os.makedirs(self.output_dir)
-        except OSError as e:
-            print(f"Error creating directory {self.output_dir}: {e}")
-            raise
+        os.makedirs(self.output_dir, exist_ok=True)
+
 
     #音声ファイルのディレクトリを受け取り、書き起こしファイルを作成する
     def create_transcription(self,segments_directory):