Use torch to determined if Whisper use GPU or CPU

app.py

@@ -6,6 +6,7 @@ import pysrt
 import pandas as pd
 from transformers import MarianMTModel, MarianTokenizer
 import ffmpeg
+import torch
 
 # Configuration initiale et chargement des données
 url = "https://huggingface.co/Lenylvt/LanguageISO/resolve/main/iso.md"
@@ -77,8 +78,8 @@ def translate_srt(input_file_path, source_language_code, target_language_code, p
 
 # Fonction pour transcrire l'audio d'une vidéo en texte
 def transcribe(audio_file_path, model_size="base"):
-    device = "cpu"
-    compute_type = "int8"
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    compute_type = "float16" if device == "cuda" else "int8"
     model = WhisperModel(model_size, device=device, compute_type=compute_type)
     segments, _ = model.transcribe(audio_file_path)
     transcription_with_timestamps = [