Update app.py

app.py

@@ -1,3 +1,4 @@
+import torch
 import gradio as gr
 from transformers import pipeline
 
@@ -5,6 +6,8 @@ MODEL_NAME = "rngzhi/cs3264-project"
 BATCH_SIZE = 8
 FILE_LIMIT_MB = 1000
 
+device = 0 if torch.cuda.is_available() else "cpu"
+
 pipe = pipeline(
     task="automatic-speech-recognition",
     model=MODEL_NAME,