Amanpreet commited on
Commit
ef58188
·
1 Parent(s): 4e2aa43
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -53,8 +53,11 @@ def run_command(command, working_dir, progress_bar, progress_text, step_start_pr
53
  try:
54
  progress = float(line.strip().split("PROGRESS:")[1]) / 100
55
  if Path(command[1]).name == 'gen_skes.py':
56
- # Scale 0-200% to 0-80%
57
- adjusted_progress = step_start_progress + (progress / 2) * step_weight
 
 
 
58
  else:
59
  # Scale 0-100% to 20% for conver_bvh.py
60
  adjusted_progress = step_start_progress + (progress * step_weight)
 
53
  try:
54
  progress = float(line.strip().split("PROGRESS:")[1]) / 100
55
  if Path(command[1]).name == 'gen_skes.py':
56
+ # Scale 0-100% (2D) to 0-60%, 100-200% (3D) to 60-80%
57
+ if progress <= 1.0:
58
+ adjusted_progress = step_start_progress + (progress * 0.6) # 60% for 2D
59
+ else:
60
+ adjusted_progress = step_start_progress + 0.6 + ((progress - 1.0) * 0.2) # 20% for 3D
61
  else:
62
  # Scale 0-100% to 20% for conver_bvh.py
63
  adjusted_progress = step_start_progress + (progress * step_weight)