MeYourHint commited on
Commit
6c5de49
·
1 Parent(s): 29b0148
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -116,7 +116,7 @@ DEFAULT_TEXT = "A person is "
116
 
117
  if not os.path.exists("checkpoints/t2m"):
118
  os.system("bash prepare/download_models_demo.sh")
119
-
120
  ##########################
121
  ######Preparing demo######
122
  ##########################
@@ -213,7 +213,7 @@ def generate(
213
  pred_motions = pred_motions.detach().cpu().numpy()
214
  data = inv_transform(pred_motions)
215
  for k, (caption, joint_data) in enumerate(zip(captions, data)):
216
- animation_path = pjoin(cached_dir, uid, str(k))
217
  os.makedirs(animation_path, exist_ok=True)
218
  joint_data = joint_data[:m_length[k]]
219
  joint = recover_from_ric(torch.from_numpy(joint_data).float(), 22).numpy()
 
116
 
117
  if not os.path.exists("checkpoints/t2m"):
118
  os.system("bash prepare/download_models_demo.sh")
119
+
120
  ##########################
121
  ######Preparing demo######
122
  ##########################
 
213
  pred_motions = pred_motions.detach().cpu().numpy()
214
  data = inv_transform(pred_motions)
215
  for k, (caption, joint_data) in enumerate(zip(captions, data)):
216
+ animation_path = pjoin(cached_dir, f'{uid}', str(k))
217
  os.makedirs(animation_path, exist_ok=True)
218
  joint_data = joint_data[:m_length[k]]
219
  joint = recover_from_ric(torch.from_numpy(joint_data).float(), 22).numpy()