Update app.py

app.py

@@ -1,3 +1,5 @@
+import torch
+
 import gradio as gr
 import pytube as pt
 from transformers import pipeline
@@ -6,7 +8,7 @@ from huggingface_hub import model_info
 MODEL_NAME = "thomas0104/whisper-large-v2-nan-tw-only-char" #this always needs to stay in line 8 :D sorry for the hackiness
 lang = "chinese"
 
-
+device = 0 if torch.cuda.is_available() else "cpu"
 pipe = pipeline(
     task="automatic-speech-recognition",
     model=MODEL_NAME,