Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
a0da0d6
1
Parent(s):
eea070b
docs: little update
Browse files- app.py +2 -1
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -613,7 +613,7 @@ def inference(inference_canvas, visual_canvas, text_prompt, traj_lists, main_ref
|
|
| 613 |
|
| 614 |
|
| 615 |
# Create the traj tensor
|
| 616 |
-
print("inference_canvas is", inference_canvas)
|
| 617 |
traj_tensor, traj_imgs_np, _, img_with_traj = VideoDataset_Motion.prepare_traj_tensor(
|
| 618 |
full_pred_tracks, canvas_height, canvas_width,
|
| 619 |
[], dot_radius, canvas_width, canvas_height,
|
|
@@ -623,6 +623,7 @@ def inference(inference_canvas, visual_canvas, text_prompt, traj_lists, main_ref
|
|
| 623 |
|
| 624 |
# Store Trajectory
|
| 625 |
imageio.mimsave(os.path.join(store_folder_path, "traj_video.mp4"), traj_imgs_np, fps=8)
|
|
|
|
| 626 |
|
| 627 |
######################################################################################################################################################
|
| 628 |
|
|
|
|
| 613 |
|
| 614 |
|
| 615 |
# Create the traj tensor
|
| 616 |
+
print("inference_canvas shape is", inference_canvas.shape)
|
| 617 |
traj_tensor, traj_imgs_np, _, img_with_traj = VideoDataset_Motion.prepare_traj_tensor(
|
| 618 |
full_pred_tracks, canvas_height, canvas_width,
|
| 619 |
[], dot_radius, canvas_width, canvas_height,
|
|
|
|
| 623 |
|
| 624 |
# Store Trajectory
|
| 625 |
imageio.mimsave(os.path.join(store_folder_path, "traj_video.mp4"), traj_imgs_np, fps=8)
|
| 626 |
+
print("Save Traj video to tmp folder!")
|
| 627 |
|
| 628 |
######################################################################################################################################################
|
| 629 |
|
requirements.txt
CHANGED
|
@@ -13,7 +13,7 @@ sentencepiece
|
|
| 13 |
# qwen-vl-utils[decord]==0.0.8
|
| 14 |
scikit-learn
|
| 15 |
matplotlib
|
| 16 |
-
gradio
|
| 17 |
imageio-ffmpeg
|
| 18 |
git+https://github.com/facebookresearch/segment-anything.git
|
| 19 |
accelerate
|
|
|
|
| 13 |
# qwen-vl-utils[decord]==0.0.8
|
| 14 |
scikit-learn
|
| 15 |
matplotlib
|
| 16 |
+
gradio==5.49.1
|
| 17 |
imageio-ffmpeg
|
| 18 |
git+https://github.com/facebookresearch/segment-anything.git
|
| 19 |
accelerate
|