Spaces:
Runtime error
Runtime error
Commit
·
69c91fa
1
Parent(s):
02851cf
fix
Browse files
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)
|