Samuel L Meyers commited on
Commit
8688550
·
1 Parent(s): 96b4603

Hopefully fixed app, even if it's slow.

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -4,15 +4,14 @@ import scipy
4
  import torch
5
 
6
  processor = AutoProcessor.from_pretrained("suno/bark-small")
7
- model = BarkModel.from_pretrained("suno/bark-small", torch_dtype=torch.float16, use_flash_attention_2=True)
8
- model = model.to_bettertransformer()
9
 
10
  def greet(text):
11
  inputs = processor(
12
  text=[text],
13
  return_tensors="pt",
14
  )
15
- speech_values = model.generate(**inputs, do_sample=True)
16
  scipy.io.wavfile.write("tmp.wav", rate=24000, data=speech_values.cpu().numpy().squeeze())
17
  return open("tmp.wav", "rb").read()
18
 
 
4
  import torch
5
 
6
  processor = AutoProcessor.from_pretrained("suno/bark-small")
7
+ model = BarkModel.from_pretrained("suno/bark-small")
 
8
 
9
  def greet(text):
10
  inputs = processor(
11
  text=[text],
12
  return_tensors="pt",
13
  )
14
+ speech_values = model.generate(**inputs, do_sample=True, num_beams=5)
15
  scipy.io.wavfile.write("tmp.wav", rate=24000, data=speech_values.cpu().numpy().squeeze())
16
  return open("tmp.wav", "rb").read()
17