Update app.py

app.py

@@ -70,13 +70,13 @@ def process(text_query, full_video_path):
         subclip = video.subclip(start_pt, end_pt)
         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)
+    # checkpoint_path ='./refer-youtube-vos_window-12.pth.tar'
+    model, postprocessor = torch.hub.load('Randl/MTTR:main','mttr_refer_youtube_vos', get_weights=True)
     
-    model_state_dict = torch.load(checkpoint_path, map_location='cpu')
-    if 'model_state_dict' in model_state_dict.keys():
-        model_state_dict = model_state_dict['model_state_dict']
-    model.load_state_dict(model_state_dict, strict=True)
+    # model_state_dict = torch.load(checkpoint_path, map_location='cpu')
+    # if 'model_state_dict' in model_state_dict.keys():
+    #     model_state_dict = model_state_dict['model_state_dict']
+    # model.load_state_dict(model_state_dict, strict=True)
 
 
     text_queries= [text_query]