H-Liu1997 commited on
Commit
3b49069
1 Parent(s): d14cfd2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -3
app.py CHANGED
@@ -602,7 +602,11 @@ class BaseTrainer(object):
602
  use_matplotlib = False,
603
  args = self.args,
604
  )
605
- result = gr.Video(value=render_vid_path, visible=True)
 
 
 
 
606
  end_time = time.time() - start_time
607
  logger.info(f"total inference time: {int(end_time)} s for {int(total_length/self.args.pose_fps)} s motion")
608
  return result
@@ -642,7 +646,10 @@ demo = gr.Interface(
642
  gr.Audio(),
643
  # gr.File(label="Please upload textgrid format file here.", file_types=["TextGrid", "Textgrid", "textgrid"])
644
  ], # input type
645
- outputs=gr.Video(format="mp4", visible=True),
 
 
 
646
  title='\
647
  <div align="center">\
648
  EMAGE: Towards Unified Holistic Co-Speech Gesture Generation via Expressive Masked Audio Gesture Modeling <br/>\
@@ -658,7 +665,8 @@ demo = gr.Interface(
658
  </div>\
659
  ',
660
  article="\
661
- Due to the limited resources in this space, we process the first 60s of your uploaded audio,try to develop this space locally for longer motion generation, [Project Page](https://pantomatrix.github.io/EMAGE/)\
 
662
  ",
663
  examples=examples,
664
  )
 
602
  use_matplotlib = False,
603
  args = self.args,
604
  )
605
+ result = [
606
+ gr.Video(value=render_vid_path, visible=True),
607
+ gr.File(value=results_save_path+"res_"+test_seq_list.iloc[its]['id']+'.npz', visible=True),
608
+ ]
609
+
610
  end_time = time.time() - start_time
611
  logger.info(f"total inference time: {int(end_time)} s for {int(total_length/self.args.pose_fps)} s motion")
612
  return result
 
646
  gr.Audio(),
647
  # gr.File(label="Please upload textgrid format file here.", file_types=["TextGrid", "Textgrid", "textgrid"])
648
  ], # input type
649
+ outputs=[
650
+ gr.Video(format="mp4", visible=True),
651
+ gr.File(label="download motion and visualize in blender"),
652
+ ],
653
  title='\
654
  <div align="center">\
655
  EMAGE: Towards Unified Holistic Co-Speech Gesture Generation via Expressive Masked Audio Gesture Modeling <br/>\
 
665
  </div>\
666
  ',
667
  article="\
668
+ For appling motion on your avatar: download npz file and blender v3.3 add-on on our project page, then retarget the motion. <br/> \
669
+ Due to the limited resources in this space, we process the first 60s of your uploaded audio,try to develop this space locally for longer motion generation, \[Project Page](https://pantomatrix.github.io/EMAGE/)\
670
  ",
671
  examples=examples,
672
  )