Update inference/svs/gradio/infer.py

inference/svs/gradio/infer.py

@@ -10,6 +10,7 @@ from utils.hparams import set_hparams
 from utils.hparams import hparams as hp
 import numpy as np
 
+import spaces
 
 class GradioInfer:
     def __init__(self, exp_name, inference_cls, title, description, article, example_inputs):
@@ -22,6 +23,7 @@ class GradioInfer:
         cls_name = inference_cls.split(".")[-1]
         self.inference_cls = getattr(importlib.import_module(pkg), cls_name)
 
+    @spaces.GPU(duration=180)
     def greet(self, text, notes, notes_duration):
         PUNCS = '。？；：'
         sents = re.split(rf'([{PUNCS}])', text.replace('\n', ','))