Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -445,9 +445,6 @@ checkpoint_fodler = os.path.join('.', 'weights')
|
|
| 445 |
|
| 446 |
sam_checkpoint = load_file_from_url(sam_checkpoint_url_dict[args.sam_model_type], checkpoint_fodler)
|
| 447 |
cutie_checkpoint = load_file_from_url(os.path.join(pretrain_model_url, 'cutie-base-mega.pth'), checkpoint_fodler)
|
| 448 |
-
# propainter_checkpoint = load_file_from_url(os.path.join(pretrain_model_url, 'ProPainter.pth'), checkpoint_fodler)
|
| 449 |
-
# raft_checkpoint = load_file_from_url(os.path.join(pretrain_model_url, 'raft-things.pth'), checkpoint_fodler)
|
| 450 |
-
# flow_completion_checkpoint = load_file_from_url(os.path.join(pretrain_model_url, 'recurrent_flow_completion.pth'), checkpoint_fodler)
|
| 451 |
|
| 452 |
# initialize sam, cutie, propainter models
|
| 453 |
model = TrackingAnything(sam_checkpoint, cutie_checkpoint, args)
|
|
@@ -457,7 +454,7 @@ title = r"""<h1 align="center">ROSE: Remove Objects with Side Effects in Videos<
|
|
| 457 |
|
| 458 |
description = r"""
|
| 459 |
<center></center>
|
| 460 |
-
<b>Official Gradio demo</b> for <a href='https://github.com/
|
| 461 |
π₯ ROSE is a robust inpainting algorithm.<br>
|
| 462 |
π€ Try to drop your video, add the masks and get the the inpainting results!<br>
|
| 463 |
"""
|
|
|
|
| 445 |
|
| 446 |
sam_checkpoint = load_file_from_url(sam_checkpoint_url_dict[args.sam_model_type], checkpoint_fodler)
|
| 447 |
cutie_checkpoint = load_file_from_url(os.path.join(pretrain_model_url, 'cutie-base-mega.pth'), checkpoint_fodler)
|
|
|
|
|
|
|
|
|
|
| 448 |
|
| 449 |
# initialize sam, cutie, propainter models
|
| 450 |
model = TrackingAnything(sam_checkpoint, cutie_checkpoint, args)
|
|
|
|
| 454 |
|
| 455 |
description = r"""
|
| 456 |
<center></center>
|
| 457 |
+
<b>Official Gradio demo</b> for <a href='https://github.com/Kunbyte-AI/ROSE' target='_blank'><b>Remove Objects with Side Effects in Videos</b></a>.<br>
|
| 458 |
π₯ ROSE is a robust inpainting algorithm.<br>
|
| 459 |
π€ Try to drop your video, add the masks and get the the inpainting results!<br>
|
| 460 |
"""
|