Update app_demo.py

app_demo.py

@@ -22,6 +22,8 @@ import numpy as np
 import torch
 import random
 
+import spaces
+
 import logging
 logging.getLogger('matplotlib').setLevel(logging.WARNING)
 
@@ -62,6 +64,7 @@ instruct_dict = {'预训练音色': '1. 选择预训练音色\n2.点击生成音
 def change_instruction(mode_checkbox_group):
     return instruct_dict[mode_checkbox_group]
 
+@spaces.GPU
 def generate_audio(tts_text, mode_checkbox_group, sft_dropdown, prompt_text, prompt_wav_upload, prompt_wav_record, instruct_text, seed):
     return "jay_short.wav"