Xu Xuenan commited on
Commit
d0fc9a1
·
1 Parent(s): 6fc0fd0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -85,11 +85,12 @@ def write_story_fn(story_topic, main_role, scene,
85
  # story_data, story_accordion, story_content
86
  return pages, gr.update(visible=True), pages[current_page], gr.update()
87
 
88
- @spaces.GPU
89
  def modality_assets_generation_fn(
90
  height, width, image_seed, sound_guidance_scale, sound_seed,
91
  n_candidate_per_text, music_duration,
92
- story_data):
 
93
  deep_update(config, {
94
  "image_generation": {
95
  "obj_cfg": {
 
85
  # story_data, story_accordion, story_content
86
  return pages, gr.update(visible=True), pages[current_page], gr.update()
87
 
88
+ @spaces.GPU(duration=600)
89
  def modality_assets_generation_fn(
90
  height, width, image_seed, sound_guidance_scale, sound_seed,
91
  n_candidate_per_text, music_duration,
92
+ story_data,
93
+ progress=gr.Progress(track_tqdm=True)):
94
  deep_update(config, {
95
  "image_generation": {
96
  "obj_cfg": {