fffiloni commited on
Commit
c6fe956
·
1 Parent(s): a31a724

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -26,8 +26,8 @@ def infer(text_prompt):
26
 
27
  with torch.inference_mode():
28
  speech_output = model.generate(**inputs, do_sample = True, fine_temperature = 0.4, coarse_temperature = 0.8)
29
- audio_out = speech_output[0].cpu().numpy()
30
-
31
  # Assuming audio_out contains audio data and the sampling rate
32
  sampling_rate = model.generation_config.sample_rate
33
  wavfile.write("output.wav", sampling_rate, audio_out)
 
26
 
27
  with torch.inference_mode():
28
  speech_output = model.generate(**inputs, do_sample = True, fine_temperature = 0.4, coarse_temperature = 0.8)
29
+ #audio_out = speech_output[0].cpu().numpy()
30
+ audio_out = speech_output[0]
31
  # Assuming audio_out contains audio data and the sampling rate
32
  sampling_rate = model.generation_config.sample_rate
33
  wavfile.write("output.wav", sampling_rate, audio_out)