move to device

app.py

@@ -2,7 +2,7 @@ import os
 
 import gradio as gr
 import torch
-from styletts2importable import compute_style, inference
+from styletts2importable import compute_style, device, inference
 from txtsplit import txtsplit
 import numpy as np
 import phonemizer
@@ -35,7 +35,7 @@ global_phonemizer = phonemizer.backend.EspeakBackend(
 for v in voicelist:
     cache_path = f"voices/{v}.wav.npy"
     if os.path.exists(cache_path):
-        voices[v] = torch.from_numpy(np.load(cache_path))
+        voices[v] = torch.from_numpy(np.load(cache_path)).to(device)
     else:
         style = compute_style(f"voices/{v}.wav")
         voices[v] = style