hf fix3

Dockerfile

@@ -9,6 +9,8 @@ WORKDIR /app
 COPY requirements.txt .
 RUN pip3 install --no-cache-dir --upgrade -r requirements.txt
 
+RUN python3 -c "from faster_whisper import WhisperModel; WhisperModel.download_model('small', cache_dir='/app/models')"
+
 COPY . .
 
 EXPOSE 7860

app.py

@@ -214,12 +214,16 @@ async def generate_subtitles(
         file_name_without_extension, _ = os.path.splitext(base_name)
 
         FILE_NAME_FOR_TXT = file_name_without_extension
-        MODEL_NAME = "small"
+        model_size = "small"
 
-        print(f"\nLoading Whisper model: {MODEL_NAME}...")
+        print(f"\nLoading Whisper model: {model_size}...")
         try:
-            model_path = os.path.join('models', 'snapshots', '536b0662742c02347bc0e980a01041f333bce120')
-            model = WhisperModel(model_path, device="cpu", compute_type="int8")
+            model = WhisperModel(
+                model_size,
+                device="cpu",
+                compute_type="int8",
+                download_root="/app/models"
+            )
             batched_model = BatchedInferencePipeline(model=model)
             print("Model loaded successfully.")