Update app.py

app.py

@@ -1,5 +1,5 @@
 import spaces
-
+import torch
 import gradio as gr
 import yt_dlp as youtube_dl
 from transformers import pipeline
@@ -13,7 +13,7 @@ BATCH_SIZE = 8
 FILE_LIMIT_MB = 1000
 YT_LENGTH_LIMIT_S = 3600  # limit to 1 hour YouTube files
 
-device = 0
+device = 0 if torch.cuda.is_available() else "cpu"
 
 pipe = pipeline(
     task="automatic-speech-recognition",