Add device_dict

app.py

@@ -16,6 +16,7 @@ TRANSCRIBER_MODEL_NAME = "juancopi81/whisper-medium-es"
 lang = "es"
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
+device_dict = {"cuda": 0, "cpu": -1}
 dtype = torch.float16 if device == "cuda" else torch.float32
 
 # Detect if code is running in Colab
@@ -32,7 +33,7 @@ transcription_pipe = pipeline(
     task="automatic-speech-recognition",
     model=TRANSCRIBER_MODEL_NAME,
     chunk_length_s=30,
-    device=device,
+    device=device_dict[device],
 )
 transcription_pipe.model.config.forced_decoder_ids = transcription_pipe.tokenizer.get_decoder_prompt_ids(language=lang, task="transcribe")
 audio_transcriber = Transcriber(transcription_pipe)