add huggingface gpu

app.py

@@ -6,6 +6,7 @@ import soundfile as sf
 import numpy as np
 import torch
 import traceback
+import spaces
 
 repo_url = "https://huggingface.co/dangtr0408/StyleTTS2-lite-vi"
 repo_dir = "StyleTTS2-lite-vi"
@@ -27,6 +28,7 @@ eg_texts = [
 
 
 # Core inference function
+@spaces.GPU
 def main(reference_paths, text_prompt, denoise, avg_style, stabilize):
     try:
         speakers = {}