Update app.py
Browse files
app.py
CHANGED
@@ -1483,7 +1483,10 @@ def bark_tts(text):
|
|
1483 |
print(f"🔁 Synthesizing TTS for: {text}")
|
1484 |
inputs = processor_bark(text, return_tensors="pt", voice_preset=bark_voice_preset)
|
1485 |
inputs = {k: v.to(model_bark.device) for k, v in inputs.items()}
|
1486 |
-
|
|
|
|
|
|
|
1487 |
speech = speech_values.cpu().numpy().squeeze()
|
1488 |
speech = (speech * 32767).astype(np.int16)
|
1489 |
temp_wav = tempfile.NamedTemporaryFile(delete=False, suffix=".wav")
|
|
|
1483 |
print(f"🔁 Synthesizing TTS for: {text}")
|
1484 |
inputs = processor_bark(text, return_tensors="pt", voice_preset=bark_voice_preset)
|
1485 |
inputs = {k: v.to(model_bark.device) for k, v in inputs.items()}
|
1486 |
+
inputs["max_new_tokens"] = 100 # Add this to the input dictionary
|
1487 |
+
start = time.time()
|
1488 |
+
speech_values = model_bark.generate(**inputs)
|
1489 |
+
print(f"✅ Bark finished in {round(time.time() - start, 2)}s")
|
1490 |
speech = speech_values.cpu().numpy().squeeze()
|
1491 |
speech = (speech * 32767).astype(np.int16)
|
1492 |
temp_wav = tempfile.NamedTemporaryFile(delete=False, suffix=".wav")
|