Update app.py
Browse files
app.py
CHANGED
|
@@ -76,7 +76,7 @@ def infer(text_prompt):
|
|
| 76 |
def generate(text_prompt, out_type):
|
| 77 |
text_prompt = text_prompt
|
| 78 |
|
| 79 |
-
inputs = processor(text_prompt, voice_preset="
|
| 80 |
|
| 81 |
with torch.inference_mode():
|
| 82 |
speech_output = model.generate(**inputs)
|
|
|
|
| 76 |
def generate(text_prompt, out_type):
|
| 77 |
text_prompt = text_prompt
|
| 78 |
|
| 79 |
+
inputs = processor(text_prompt, voice_preset="v2/en_speaker_6").to(device)
|
| 80 |
|
| 81 |
with torch.inference_mode():
|
| 82 |
speech_output = model.generate(**inputs)
|