yasserrmd commited on
Commit
beb79e1
1 Parent(s): 348b950

Update generate_audio.py

Browse files
Files changed (1) hide show
  1. generate_audio.py +4 -4
generate_audio.py CHANGED
@@ -43,10 +43,10 @@ class TTSGenerator:
43
  """
44
 
45
  # Load Bark model and processor for Speaker 2
46
- self.bark_processor = AutoProcessor.from_pretrained("suno/bark-small")
47
- self.bark_model = BarkModel.from_pretrained("suno/bark-small", torch_dtype=torch.float16).to(self.device)
48
- self.bark_sampling_rate = 24000
49
- self.voice_preset = "v2/en_speaker_6"
50
 
51
  @spaces.GPU
52
  def load_transcript(self):
 
43
  """
44
 
45
  # Load Bark model and processor for Speaker 2
46
+ # self.bark_processor = AutoProcessor.from_pretrained("suno/bark-small")
47
+ # self.bark_model = BarkModel.from_pretrained("suno/bark-small", torch_dtype=torch.float16).to(self.device)
48
+ # self.bark_sampling_rate = 24000
49
+ # self.voice_preset = "v2/en_speaker_6"
50
 
51
  @spaces.GPU
52
  def load_transcript(self):