husseinelsaadi commited on
Commit
a6c48a9
·
verified ·
1 Parent(s): c343ad6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1483,7 +1483,7 @@ 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)
1487
  speech = speech_values.cpu().numpy().squeeze()
1488
  speech = (speech * 32767).astype(np.int16)
1489
  temp_wav = tempfile.NamedTemporaryFile(delete=False, suffix=".wav")
@@ -1747,10 +1747,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
1747
  )
1748
  confirm_btn.click(
1749
  process_answer,
1750
- [stt_transcript, user_audio_input, user_video_input, interview_state, user_data],
1751
- [interview_state, interview_summary, question_audio, question_text, user_audio_input, user_video_input, emotion_display]
1752
  ).then(
1753
- lambda: (gr.update(value=None), gr.update(value=None)), None, [user_audio_input, user_video_input]
1754
  )
1755
 
1756
  demo.launch(debug=True)
 
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=300)
1487
  speech = speech_values.cpu().numpy().squeeze()
1488
  speech = (speech * 32767).astype(np.int16)
1489
  temp_wav = tempfile.NamedTemporaryFile(delete=False, suffix=".wav")
 
1747
  )
1748
  confirm_btn.click(
1749
  process_answer,
1750
+ [stt_transcript, user_audio_input, , interview_state, user_data],
1751
+ [interview_state, interview_summary, question_audio, question_text, user_audio_input]
1752
  ).then(
1753
+ lambda: (gr.update(value=None), gr.update(value=None)), None, [user_audio_input]
1754
  )
1755
 
1756
  demo.launch(debug=True)