MeYourHint commited on
Commit
055110b
·
1 Parent(s): abcf2c4
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -217,6 +217,7 @@ def generate(
217
  data = inv_transform(pred_motions)
218
  for k, (caption, joint_data) in enumerate(zip(captions, data)):
219
  animation_path = pjoin(cached_dir, f'{uid}')
 
220
  os.makedirs(animation_path, exist_ok=True)
221
  joint_data = joint_data[:m_length[k]]
222
  joint = recover_from_ric(torch.from_numpy(joint_data).float(), 22).numpy()
 
217
  data = inv_transform(pred_motions)
218
  for k, (caption, joint_data) in enumerate(zip(captions, data)):
219
  animation_path = pjoin(cached_dir, f'{uid}')
220
+ shutil.rmtree(animation_path)
221
  os.makedirs(animation_path, exist_ok=True)
222
  joint_data = joint_data[:m_length[k]]
223
  joint = recover_from_ric(torch.from_numpy(joint_data).float(), 22).numpy()