Spaces:

malvin-ai
/

light-ai-video-generator

Running on Zero

malvin noel commited on Apr 24

Commit

7f0f936

1 Parent(s): 8ba85e6

correct import whisper cpu

Files changed (1) hide show

scripts/generate_subtitles.py CHANGED Viewed

@@ -89,7 +89,7 @@ def transcribe_audio_to_subs(audio_path):
     """
     print("🎙️ Transcription avec Whisper...")
     import whisper
-    model = whisper.load_model("medium")
     result = model.transcribe(audio_path)
     subtitles = [{

     """
     print("🎙️ Transcription avec Whisper...")
     import whisper
+    model = whisper.load_model("medium", device="cpu")
     result = model.transcribe(audio_path)
     subtitles = [{