Update app.py

app.py

@@ -8,7 +8,7 @@ import gradio as gr
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 pipe_dict = {
-    "original_pipe": pipeline("text-to-speech", model="kakao-enterprise/vits-ljs", device=0),
+    "original_pipe": pipeline("text-to-speech", model="kakao-enterprise/vits-ljs", device=device),
 }
 
 # Inference