Remy commited on
Commit
4fa464a
·
verified ·
1 Parent(s): 700229e

Update ActionMesh space

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -330,7 +330,7 @@ def _run_actionmesh_impl(
330
 
331
  try:
332
  # Determine input source
333
- progress(None, desc="Loading input...")
334
 
335
  if video_input is not None:
336
  input_path = video_input
@@ -352,7 +352,7 @@ def _run_actionmesh_impl(
352
  return None, None, None, "Error: At least 16 frames are required."
353
 
354
  # Get pipeline and move to GPU
355
- progress(None, desc="Moving pipeline to GPU...")
356
  pipe = load_pipeline_cpu()
357
  pipe.to("cuda")
358
 
@@ -361,7 +361,7 @@ def _run_actionmesh_impl(
361
  torch.cuda.empty_cache()
362
 
363
  # Run inference
364
- progress(None, desc="Starting pipeline...")
365
 
366
  # Set steps based on quality mode
367
  if quality_mode == "⚡ Fast":
 
330
 
331
  try:
332
  # Determine input source
333
+ progress(0.0, desc="Loading input...")
334
 
335
  if video_input is not None:
336
  input_path = video_input
 
352
  return None, None, None, "Error: At least 16 frames are required."
353
 
354
  # Get pipeline and move to GPU
355
+ progress(0.0, desc="Moving pipeline to GPU...")
356
  pipe = load_pipeline_cpu()
357
  pipe.to("cuda")
358
 
 
361
  torch.cuda.empty_cache()
362
 
363
  # Run inference
364
+ progress(0.0, desc="Starting pipeline...")
365
 
366
  # Set steps based on quality mode
367
  if quality_mode == "⚡ Fast":