Update app.py

app.py

@@ -132,9 +132,6 @@ LANGUAGE_NAME_TO_CODE = {
 # Reverse mapping of language codes to full language names
 CODE_TO_LANGUAGE_NAME = {v: k for k, v in LANGUAGE_NAME_TO_CODE.items()}
 
-# Device and compute type for faster-whisper
-device, torch_dtype = ("cuda", "float32") if torch.cuda.is_available() else ("cpu", "int8")
-
 def detect_language(audio_file):
     """Detect the language of the audio file."""
     # Load the Whisper model (use "base" for faster detection)
@@ -186,8 +183,12 @@ def transcribe_audio(audio_file, language="Auto Detect", model_size="Base (Faste
     else:
         # Use the selected Whisper model
         if model_size == "Systran Faster Whisper Large v3":
+            # Define device and compute type for faster-whisper
+            device = "cuda" if torch.cuda.is_available() else "cpu"
+            compute_type = "float32" if device == "cuda" else "int8"
+            
             # Use faster-whisper for the Systran model
-            model = WhisperModel(MODELS[model_size], device=device, compute_type=torch_dtype)
+            model = WhisperModel(MODELS[model_size], device=device, compute_type=compute_type)
             segments, info = model.transcribe(
                 processed_audio_path,
                 task="transcribe",