Update app.py

app.py

@@ -162,7 +162,7 @@ def generate_audio_and_image(story_prompt, voice_preset="Speaker 6 (en)"):
     
     pieces = []
     for i in range(0, len(model_input), BATCH_SIZE):
-        inputs = model_input[BATCH_SIZE*i:min(BATCH_SIZE*(i+1), len(model_input))]
+        inputs = model_input[i:min(i + BATCH_SIZE, len(model_input))]
         
         if len(inputs) != 0:
             inputs = processor(inputs, voice_preset=voice_presets_dict[voice_preset])