Spaces:
Build error
Build error
Update ActionMesh space
Browse files
app.py
CHANGED
|
@@ -330,7 +330,7 @@ def _run_actionmesh_impl(
|
|
| 330 |
|
| 331 |
try:
|
| 332 |
# Determine input source
|
| 333 |
-
progress(
|
| 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(
|
| 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(
|
| 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":
|