Besar commited on
Commit
b7c80c9
·
1 Parent(s): ec65b3f
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -563,12 +563,12 @@ with gr.Blocks() as iface:
563
 
564
  download_video_button.click(
565
  fn=download_video,
566
- inputs=[video_input],outputs=[video_path], api_name="/api_download_video"
567
  )
568
 
569
  process_video_button.click(
570
  fn=process_video,
571
- inputs=[video_input],outputs=[video_path], api_name="/api_process_video"
572
  )
573
 
574
  # first step: get the video information
@@ -581,7 +581,7 @@ with gr.Blocks() as iface:
581
  outputs=[video_state, video_info, template_frame, image_selection_slider,
582
  track_pause_number_slider,point_prompt, clear_button_click, Add_mask_button,
583
  template_frame, tracking_video_predict_button, video_output, mask_dropdown,
584
- remove_mask_button, inpaint_video_predict_button, run_status], api_name="/api_extract_frames"
585
  )
586
 
587
  extract_frames_button.click(
@@ -592,19 +592,19 @@ with gr.Blocks() as iface:
592
  outputs=[video_state, video_info, template_frame, image_selection_slider,
593
  track_pause_number_slider,point_prompt, clear_button_click, Add_mask_button,
594
  template_frame, tracking_video_predict_button, video_output, mask_dropdown,
595
- remove_mask_button, inpaint_video_predict_button, run_status], api_name="/extract_frames"
596
  )
597
 
598
  # second step: select images from slider
599
  image_selection_slider.release(fn=select_template,
600
  inputs=[image_selection_slider, video_state, interactive_state],
601
- outputs=[template_frame, video_state, interactive_state, run_status], api_name="/select_image")
602
  track_pause_number_slider.release(fn=get_end_number,
603
  inputs=[track_pause_number_slider, video_state, interactive_state],
604
- outputs=[template_frame, interactive_state, run_status], api_name="/end_image")
605
  resize_ratio_slider.release(fn=get_resize_ratio,
606
  inputs=[resize_ratio_slider, interactive_state],
607
- outputs=[interactive_state], api_name="/resize_ratio")
608
 
609
  # click select image to get mask using sam
610
  template_frame.select(
 
563
 
564
  download_video_button.click(
565
  fn=download_video,
566
+ inputs=[video_input],outputs=[video_path], api_name="api_download_video"
567
  )
568
 
569
  process_video_button.click(
570
  fn=process_video,
571
+ inputs=[video_input],outputs=[video_path], api_name="api_process_video"
572
  )
573
 
574
  # first step: get the video information
 
581
  outputs=[video_state, video_info, template_frame, image_selection_slider,
582
  track_pause_number_slider,point_prompt, clear_button_click, Add_mask_button,
583
  template_frame, tracking_video_predict_button, video_output, mask_dropdown,
584
+ remove_mask_button, inpaint_video_predict_button, run_status], api_name="api_extract_frames"
585
  )
586
 
587
  extract_frames_button.click(
 
592
  outputs=[video_state, video_info, template_frame, image_selection_slider,
593
  track_pause_number_slider,point_prompt, clear_button_click, Add_mask_button,
594
  template_frame, tracking_video_predict_button, video_output, mask_dropdown,
595
+ remove_mask_button, inpaint_video_predict_button, run_status], api_name="extract_frames"
596
  )
597
 
598
  # second step: select images from slider
599
  image_selection_slider.release(fn=select_template,
600
  inputs=[image_selection_slider, video_state, interactive_state],
601
+ outputs=[template_frame, video_state, interactive_state, run_status], api_name="select_image")
602
  track_pause_number_slider.release(fn=get_end_number,
603
  inputs=[track_pause_number_slider, video_state, interactive_state],
604
+ outputs=[template_frame, interactive_state, run_status], api_name="end_image")
605
  resize_ratio_slider.release(fn=get_resize_ratio,
606
  inputs=[resize_ratio_slider, interactive_state],
607
+ outputs=[interactive_state], api_name="resize_ratio")
608
 
609
  # click select image to get mask using sam
610
  template_frame.select(