samadi10 commited on
Commit
9c67d49
1 Parent(s): d1290ba

feature:added video playback

Browse files
Files changed (2) hide show
  1. app.py +4 -2
  2. visualize/simplify_loc2rot.py +1 -1
app.py CHANGED
@@ -133,5 +133,7 @@ if 'html_content' in st.session_state or 'gif_base64' in st.session_state:
133
 
134
  with disp_col2:
135
  if video_path:
136
- gif_html = f'<video controls width="800" height="800"><source src="file:///{video_path}" type="video/mp4"></video autoplay>'
137
- components.html(gif_html, height=800, scrolling=True)
 
 
 
133
 
134
  with disp_col2:
135
  if video_path:
136
+ video_file = open(video_path, 'rb')
137
+ video_bytes = video_file.read()
138
+
139
+ st.video(video_bytes, format='video/mp4', loop=True)
visualize/simplify_loc2rot.py CHANGED
@@ -18,7 +18,7 @@ class joints2smpl:
18
  self.batch_size = num_frames
19
  self.num_joints = 22 # for HumanML3D
20
  self.joint_category = "AMASS"
21
- self.num_smplify_iters = 5
22
  self.fix_foot = False
23
  print(config.SMPL_MODEL_DIR)
24
  smplmodel = smplx.create(config.SMPL_MODEL_DIR,
 
18
  self.batch_size = num_frames
19
  self.num_joints = 22 # for HumanML3D
20
  self.joint_category = "AMASS"
21
+ self.num_smplify_iters = 120
22
  self.fix_foot = False
23
  print(config.SMPL_MODEL_DIR)
24
  smplmodel = smplx.create(config.SMPL_MODEL_DIR,