Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -51,7 +51,7 @@ def process(text_query, full_video_path):
|
|
51 |
subclip.write_videofile(input_clip_path)
|
52 |
|
53 |
checkpoint_path ='./refer-youtube-vos_window-12.pth.tar'
|
54 |
-
model, postprocessor = torch.hub.load('
|
55 |
|
56 |
model_state_dict = torch.load(checkpoint_path, map_location='cpu')
|
57 |
if 'model_state_dict' in model_state_dict.keys():
|
|
|
51 |
subclip.write_videofile(input_clip_path)
|
52 |
|
53 |
checkpoint_path ='./refer-youtube-vos_window-12.pth.tar'
|
54 |
+
model, postprocessor = torch.hub.load('mttr2021/MTTR:main','mttr_refer_youtube_vos', get_weights=False)
|
55 |
|
56 |
model_state_dict = torch.load(checkpoint_path, map_location='cpu')
|
57 |
if 'model_state_dict' in model_state_dict.keys():
|