modify cuda link2

scripts/generate_subtitles.py

@@ -2,6 +2,7 @@
 
 import random
 import os
+import torch
 import whisper
 from moviepy import (
     VideoFileClip,
@@ -86,7 +87,7 @@ def transcribe_audio_to_subs(audio_path):
     des segments start/end/text, et sauvegarde en .srt.
     """
     print("🎙️ Transcription avec Whisper...")
-    model = whisper.load_model("medium")  # ou "small"/"large" selon ton besoin
+    model = whisper.load_model("medium", device="cuda" if torch.cuda.is_available() else "cpu")
     result = model.transcribe(audio_path)
 
     subtitles = [{
@@ -228,10 +229,3 @@ def add_subtitles_to_video(video_path, subtitles, output_file="./assets/output/v
 
     print(f"✅ Vidéo Shorts/TikTok prête : {output_file}")
 
-# test
-if __name__ == "__main__":
-    # Exemple de test
-    video_path = "assets/backgrounds/video_only.mp4"
-    audio_path = "assets/audio/voice.mp3"
-    subtitles = transcribe_audio_to_subs(audio_path)
-    add_subtitles_to_video(video_path, subtitles, output_file="output_with_subs.mp4")