Update app.py
Browse files
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)
|