eaysu commited on
Commit
188479b
·
1 Parent(s): 0d782ce

some fixes applied

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -31,12 +31,9 @@ all_voice_presets = [
31
  def generate_speech(text, model_name, voice_preset):
32
  model = models[model_name]
33
  processor = AutoProcessor.from_pretrained(model_name) # Load processor for the selected model
34
- inputs = processor(text, voice_preset=voice_preset, return_tensors="pt")
35
- inputs['attention_mask'] = torch.ones_like(inputs['input_ids'])
36
-
37
- with torch.no_grad():
38
- audio_array = model.generate(**inputs)
39
- audio_array = audio_array.cpu().numpy().astype('float32').squeeze()
40
  return (model.generation_config.sample_rate, audio_array)
41
 
42
  # Gradio app setup
 
31
  def generate_speech(text, model_name, voice_preset):
32
  model = models[model_name]
33
  processor = AutoProcessor.from_pretrained(model_name) # Load processor for the selected model
34
+ inputs = processor(text, voice_preset=voice_preset)
35
+ audio_array = model.generate(**inputs)
36
+ audio_array = audio_array.cpu().numpy().squeeze()
 
 
 
37
  return (model.generation_config.sample_rate, audio_array)
38
 
39
  # Gradio app setup