Update app.py

app.py

@@ -2,8 +2,11 @@ import os
 from transformers import WhisperProcessor, WhisperForConditionalGeneration
 import librosa
 
+# Check if CUDA is available and set the device
+device = "cuda" if torch.cuda.is_available() else "cpu"
+print(f"Using device: {device}")
+
 # Load model and processor
-device = "cuda"
 token = os.getenv("HF_TOKEN")
 processor = WhisperProcessor.from_pretrained("jiviai/audioX-south-v1")
 model = WhisperForConditionalGeneration.from_pretrained("jiviai/audioX-south-v1").to(device)