Spaces:
fcyai
/
Running on Zero

fcyai commited on
Commit
ae8d136
1 Parent(s): 1fe26b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -5,6 +5,7 @@ import argparse
5
  import torch
6
  import gradio as gr
7
  import numpy as np
 
8
 
9
  import ChatTTS
10
 
@@ -22,6 +23,7 @@ def generate_seed():
22
  "value": new_seed
23
  }
24
 
 
25
  def generate_audio(text, temperature, top_P, top_K, audio_seed_input, text_seed_input, refine_text_flag):
26
  torch.manual_seed(audio_seed_input)
27
  rand_spk = torch.randn(768)
 
5
  import torch
6
  import gradio as gr
7
  import numpy as np
8
+ import spaces
9
 
10
  import ChatTTS
11
 
 
23
  "value": new_seed
24
  }
25
 
26
+ @spaces.GPU
27
  def generate_audio(text, temperature, top_P, top_K, audio_seed_input, text_seed_input, refine_text_flag):
28
  torch.manual_seed(audio_seed_input)
29
  rand_spk = torch.randn(768)