Update app.py
Browse files
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
|