Update app.py

app.py

@@ -5,6 +5,7 @@ from transformers.pipelines.audio_utils import ffmpeg_read
 import gradio as gr
 import gradio as gr
 
+device = 0 if torch.cuda.is_available() else "cpu"
 
 MODEL_ID = "wav2vec2-base-vinyl_condition"