husseinelsaadi commited on
Commit
83570c2
·
verified ·
1 Parent(s): 5cc76cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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
- speech_values = model_bark.generate(**inputs, max_new_tokens=100)
 
 
 
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")