Spaces:

MTTR
/

MTTR-Referring-Video-Object-Segmentation

Runtime error

MTTR commited on Dec 7, 2021

Commit

26a0c1b

1 Parent(s): 6afd9fa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -51,7 +51,7 @@ def process(text_query, full_video_path):
         subclip.write_videofile(input_clip_path)
     checkpoint_path ='./refer-youtube-vos_window-12.pth.tar'
-    model, postprocessor = torch.hub.load('Randl/MTTR:main','mttr_refer_youtube_vos', get_weights=False)
     model_state_dict = torch.load(checkpoint_path, map_location='cpu')
     if 'model_state_dict' in model_state_dict.keys():

         subclip.write_videofile(input_clip_path)
     checkpoint_path ='./refer-youtube-vos_window-12.pth.tar'
+    model, postprocessor = torch.hub.load('mttr2021/MTTR:main','mttr_refer_youtube_vos', get_weights=False)
     model_state_dict = torch.load(checkpoint_path, map_location='cpu')
     if 'model_state_dict' in model_state_dict.keys():