mshukor commited on
Commit
335bb2c
·
1 Parent(s): 402d77e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -104,9 +104,13 @@ input_args = ["", "--task=refcoco", "--beam=10", f"--path={checkpoint_path}", "-
104
  args = options.parse_args_and_arch(parser, input_args)
105
  general_cfg = convert_namespace_to_omegaconf(args)
106
  general_task = tasks.setup_task(general_cfg.task)
 
 
 
107
  general_models, general_cfg = checkpoint_utils.load_model_ensemble(
108
  utils.split_paths(general_cfg.common_eval.path),
109
- task=general_task
 
110
  )
111
 
112
  # move models to gpu
 
104
  args = options.parse_args_and_arch(parser, input_args)
105
  general_cfg = convert_namespace_to_omegaconf(args)
106
  general_task = tasks.setup_task(general_cfg.task)
107
+
108
+ overrides={"video_model_path": None, "resnet_model_path": None}
109
+
110
  general_models, general_cfg = checkpoint_utils.load_model_ensemble(
111
  utils.split_paths(general_cfg.common_eval.path),
112
+ task=general_task,
113
+ arg_overrides=overrides
114
  )
115
 
116
  # move models to gpu