ollieollie commited on
Commit
3e1f112
·
verified ·
1 Parent(s): 112c36b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import random
2
  import numpy as np
3
  import torch
4
- from chatterbox.src.chatterbox.tts import ChatterboxTTS
5
  import gradio as gr
6
 
7
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
@@ -15,7 +15,7 @@ def set_seed(seed: int):
15
  np.random.seed(seed)
16
 
17
 
18
- model = ChatterboxTTS.from_pretrained(DEVICE)
19
 
20
  def generate(text, audio_prompt_path, exaggeration, pace, temperature, seed_num):
21
  if seed_num != 0:
 
1
  import random
2
  import numpy as np
3
  import torch
4
+ from chatterbox.src.orator.tts import OratorTTS
5
  import gradio as gr
6
 
7
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
15
  np.random.seed(seed)
16
 
17
 
18
+ model = OratorTTS.from_pretrained(DEVICE)
19
 
20
  def generate(text, audio_prompt_path, exaggeration, pace, temperature, seed_num):
21
  if seed_num != 0: