add gpu tags

cosyvoice/cli/cosyvoice.py

@@ -20,7 +20,7 @@ import torch
 from cosyvoice.cli.frontend import CosyVoiceFrontEnd
 from cosyvoice.cli.model import CosyVoiceModel, CosyVoice2Model
 from cosyvoice.utils.file_utils import logging
-
+import spaces
 
 class CosyVoice:
 
@@ -132,7 +132,7 @@ class CosyVoice:
             start_time = time.time()
 
 class CosyVoice2(CosyVoice):
-
+    @spaces.GPU
     def __init__(self, model_dir, load_jit=False, load_onnx=False, load_trt=False):
         instruct = True if '-Instruct' in model_dir else False
         self.model_dir = model_dir