yonigozlan HF Staff commited on
Commit
69c91fa
·
1 Parent(s): 02851cf
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -238,6 +238,7 @@ def ensure_session_for_current_model() -> None:
238
  GLOBAL_STATE.session_repo_id = desired_repo
239
 
240
 
 
241
  def init_video_session(video: str | dict) -> tuple[AppState, int, int, Image.Image, str]:
242
  """Gradio handler: load video, init session, return state, slider bounds, and first frame."""
243
  # Reset ONLY video-related fields, keep model loaded
@@ -584,10 +585,7 @@ with gr.Blocks(title="SAM2 Video (Transformers) - Interactive Segmentation") as
584
  ckpt_progress = gr.Markdown(visible=False)
585
  load_status = gr.Markdown(visible=True)
586
  reset_btn = gr.Button("Reset Session", variant="secondary")
587
- examples_list = [
588
- ["/home/ubuntu/models_implem/tennis.mp4"],
589
- ["/home/ubuntu/models_implem/tennis.mp4"],
590
- ]
591
  with gr.Column(scale=2):
592
  preview = gr.Image(label="Preview", interactive=True)
593
  frame_slider = gr.Slider(label="Frame", minimum=0, maximum=0, step=1, value=0, interactive=True)
 
238
  GLOBAL_STATE.session_repo_id = desired_repo
239
 
240
 
241
+ @spaces.GPU()
242
  def init_video_session(video: str | dict) -> tuple[AppState, int, int, Image.Image, str]:
243
  """Gradio handler: load video, init session, return state, slider bounds, and first frame."""
244
  # Reset ONLY video-related fields, keep model loaded
 
585
  ckpt_progress = gr.Markdown(visible=False)
586
  load_status = gr.Markdown(visible=True)
587
  reset_btn = gr.Button("Reset Session", variant="secondary")
588
+ examples_list = [["./tennis.mp4"]]
 
 
 
589
  with gr.Column(scale=2):
590
  preview = gr.Image(label="Preview", interactive=True)
591
  frame_slider = gr.Slider(label="Frame", minimum=0, maximum=0, step=1, value=0, interactive=True)