torch device

app.py

@@ -72,7 +72,8 @@ def transcribe(youtube_url, audio_file, whisper_model, gemini_api_key, gemini_pr
             gemini_api_key = default_gemini_api_key
         model = configure_genai(gemini_api_key, gemini_model_variant)
 
-        device = 0 if torch.cuda.is_available() else "cpu"
+        # device = 0 if torch.cuda.is_available() else "cpu"
+        device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         pipe = pipeline(
             task="automatic-speech-recognition",
             model=whisper_model,