HikariDawn commited on
Commit
a0da0d6
·
1 Parent(s): eea070b

docs: little update

Browse files
Files changed (2) hide show
  1. app.py +2 -1
  2. 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