Update app.py

app.py

@@ -7,6 +7,7 @@ import gradio as gr
 import whisper
 
 model = whisper.load_model("small")
+device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
 def inference(audio):
     audio = whisper.load_audio(audio)