weiyuchoumou526 commited on
Commit
a18f370
Β·
verified Β·
1 Parent(s): 12a64aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
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/sczhou/ProPainter' target='_blank'><b>Remove Objects with Side Effects in Videos</b></a>.<br>
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
  """