Update app.py

app.py

@@ -53,7 +53,9 @@ async def queue_api_calls(sentence_mapping, character_dict, selected_style):
 
     tasks = [model_actor.generate_image.remote(prompt, f"Prompt {paragraph_number}") for paragraph_number, prompt in prompts]
     print("Tasks created for image generation.")
-    responses = await asyncio.gather(*[ray.get(task) for task in tasks])
+
+    done_tasks, _ = await asyncio.wait([asyncio.create_task(ray.wait([task])) for task in tasks], return_when=asyncio.ALL_COMPLETED)
+    responses = [await ray.get(task) for task in done_tasks]
     print("Responses received from image generation tasks.")
 
     images = {paragraph_number: response for (paragraph_number, _), response in zip(prompts, responses)}