deekshachilukuri commited on
Commit
fa0a6d8
1 Parent(s): 552e7bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -15,6 +15,7 @@ with patch('builtins.input', always_yes):
15
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
16
  tts = TTS(model_name="tts_models/multilingual/multi-dataset/xtts_v2", progress_bar=False).to(device)
17
 
 
18
  def generate_voice(text, audio_file_path):
19
  # Create a directory if it does not exist
20
  output_dir = "/tmp/"
 
15
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
16
  tts = TTS(model_name="tts_models/multilingual/multi-dataset/xtts_v2", progress_bar=False).to(device)
17
 
18
+ @spaces.GPU(enable_queue=True)
19
  def generate_voice(text, audio_file_path):
20
  # Create a directory if it does not exist
21
  output_dir = "/tmp/"