Update app.py

app.py

@@ -90,7 +90,10 @@ def generate_base(subject, setting):
     description = "Jenny speaks at an average pace with a calm delivery in a very confined sounding environment with clear audio quality."
     story_tokens = prompt_tokenizer(model_input_tokens, return_tensors="pt", padding=True).to(device)
     description_tokens = description_tokenizer([description for _ in range(len(model_input_tokens))], return_tensors="pt").to(device)
-        speech_output = model.generate(input_ids=description_tokens.input_ids, prompt_input_ids=story_tokens.input_ids, attention_mask=description_tokens.attention_mask, prompt_attention_mask=story_tokens.attention_mask)
+    speech_output = model.generate(input_ids=description_tokens.input_ids,
+                                   prompt_input_ids=story_tokens.input_ids,
+                                   attention_mask=description_tokens.attention_mask,
+                                   prompt_attention_mask=story_tokens.attention_mask)
     speech_output = [output.cpu().numpy() for output in speech_output]
     gr.Info("Generated Audio")
     return None, None, {"audio": speech_output, "text": model_input_tokens}