Update app.py

app.py

@@ -1,3 +1,5 @@
+import spaces
+
 from pathlib import Path
 import argparse
 import shutil
@@ -83,6 +85,7 @@ def write_story_fn(story_topic, main_role, scene,
     # story_data, story_accordion, story_content
     return pages, gr.update(visible=True), pages[current_page], gr.update()
 
+@spaces.GPU
 def modality_assets_generation_fn(
         height, width, image_seed, sound_guidance_scale, sound_seed,
         n_candidate_per_text, music_duration,