Update app.py

app.py

@@ -5,7 +5,7 @@ from transformers import pipeline
 from transformers import BarkModel
 from transformers import AutoProcessor
 
-device="cpu"
+device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
 pipe = pipeline(
     "automatic-speech-recognition", model="openai/whisper-large-v2", device=device