Update app.py

app.py

@@ -197,7 +197,7 @@ def generate_story_image(prompt):
     return image
 
 @torch.inference_mode()
-@spaces.GPU(duration=180)  # Longer duration for multiple image generation
+@spaces.GPU(duration=180)
 def generate_all_scenes(prompts_text):
     generated_images = []
     formatted_prompts = []
@@ -221,10 +221,14 @@ def generate_all_scenes(prompts_text):
         
         if scene_prompt:
             try:
-                torch.cuda.empty_cache()
+                torch.cuda.empty_cache()  # Clear GPU memory before generation
+                print(f"Generating image for scene: {scene_prompt}")  # Debugging
                 image = generate_story_image(scene_prompt)
+                
                 if image is not None:
                     generated_images.append(np.array(image))
+                
+                torch.cuda.empty_cache()  # Clear memory after each image
             except Exception as e:
                 print(f"Error generating image: {str(e)}")
                 continue