Update app.py

app.py

@@ -37,6 +37,7 @@ vctk_model.to(device)
 @spaces.GPU
 def ljs_forward(text, speaking_rate=1.0):
     inputs = ljs_tokenizer(text, return_tensors="pt")
+    inputs = inputs.to(ljs_model.device)
 
     ljs_model.speaking_rate = speaking_rate
     set_seed(555)
@@ -50,6 +51,7 @@ def ljs_forward(text, speaking_rate=1.0):
 @spaces.GPU
 def vctk_forward(text, speaking_rate=1.0, speaker_id=1):
     inputs = vctk_tokenizer(text, return_tensors="pt")
+    inputs = inputs.to(vctk_model.device)
 
     vctk_model.speaking_rate = speaking_rate
     set_seed(555)