Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
189fb9e
1
Parent(s):
885c3cb
update requirements and add examples
Browse files- app.py +12 -0
- requirements.txt +1 -0
app.py
CHANGED
|
@@ -484,6 +484,9 @@ with gr.Blocks(title="SAM2 Video (Transformers) - Interactive Segmentation (CPU)
|
|
| 484 |
ckpt_progress = gr.Markdown(visible=False)
|
| 485 |
load_status = gr.Markdown(visible=True)
|
| 486 |
reset_btn = gr.Button("Reset Session", variant="secondary")
|
|
|
|
|
|
|
|
|
|
| 487 |
with gr.Column(scale=2):
|
| 488 |
preview = gr.Image(label="Preview", interactive=True)
|
| 489 |
frame_slider = gr.Slider(label="Frame", minimum=0, maximum=0, step=1, value=0, interactive=True)
|
|
@@ -508,6 +511,15 @@ with gr.Blocks(title="SAM2 Video (Transformers) - Interactive Segmentation (CPU)
|
|
| 508 |
video_in.change(
|
| 509 |
_on_video_change, inputs=[video_in], outputs=[state, frame_slider, preview, load_status], show_progress=True
|
| 510 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 511 |
|
| 512 |
def _on_ckpt_change(s: AppState, key: str):
|
| 513 |
if s is not None and key:
|
|
|
|
| 484 |
ckpt_progress = gr.Markdown(visible=False)
|
| 485 |
load_status = gr.Markdown(visible=True)
|
| 486 |
reset_btn = gr.Button("Reset Session", variant="secondary")
|
| 487 |
+
examples_list = [
|
| 488 |
+
["./tennis.mp4"],
|
| 489 |
+
]
|
| 490 |
with gr.Column(scale=2):
|
| 491 |
preview = gr.Image(label="Preview", interactive=True)
|
| 492 |
frame_slider = gr.Slider(label="Frame", minimum=0, maximum=0, step=1, value=0, interactive=True)
|
|
|
|
| 511 |
video_in.change(
|
| 512 |
_on_video_change, inputs=[video_in], outputs=[state, frame_slider, preview, load_status], show_progress=True
|
| 513 |
)
|
| 514 |
+
gr.Examples(
|
| 515 |
+
examples=examples_list,
|
| 516 |
+
inputs=[video_in],
|
| 517 |
+
fn=_on_video_change,
|
| 518 |
+
outputs=[state, frame_slider, preview, load_status],
|
| 519 |
+
label="Examples",
|
| 520 |
+
cache_examples=False,
|
| 521 |
+
examples_per_page=5,
|
| 522 |
+
)
|
| 523 |
|
| 524 |
def _on_ckpt_change(s: AppState, key: str):
|
| 525 |
if s is not None and key:
|
requirements.txt
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
gradio
|
| 2 |
git+https://github.com/SangbumChoi/transformers.git@sam2
|
| 3 |
torch
|
|
|
|
| 4 |
pillow
|
| 5 |
opencv-python
|
| 6 |
imageio[pyav]
|
|
|
|
| 1 |
gradio
|
| 2 |
git+https://github.com/SangbumChoi/transformers.git@sam2
|
| 3 |
torch
|
| 4 |
+
torchvision
|
| 5 |
pillow
|
| 6 |
opencv-python
|
| 7 |
imageio[pyav]
|