Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import torch
 from accelerate import init_empty_weights
 import random
 import json
@@ -19,7 +20,7 @@ genders = sorted(set(entry["gender"] for entry in data))
 accents = sorted(set(entry["accent"] for entry in data))
 
 # Load ASR pipelines
-device = 0
+device = 0 if torch.cuda.is_available() else -1
 pipe_whisper_tiny = pipeline("automatic-speech-recognition", model="openai/whisper-tiny", device=device)
 pipe_whisper_tiny_en = pipeline("automatic-speech-recognition", model="openai/whisper-tiny.en", device=device)
 pipe_whisper_base = pipeline("automatic-speech-recognition", model="openai/whisper-base", device=device)