Update app.py
Browse files
app.py
CHANGED
@@ -92,7 +92,7 @@ caption_models, caption_cfg, caption_task = checkpoint_utils.load_model_ensemble
|
|
92 |
)
|
93 |
|
94 |
# Load ckpt & config for Video Captioning
|
95 |
-
checkpoint_path = 'checkpoints/unival_video_caption_stage_1/checkpoint_best.
|
96 |
|
97 |
caption_overrides={"eval_cider":False, "beam":5, "max_len_b":22, "no_repeat_ngram_size":3, "seed":7, "unnormalized": False,
|
98 |
"bpe_dir":"utils/BPE", "video_model_path": None, "video_model_path": None, "resnet_model_path": None}
|
|
|
92 |
)
|
93 |
|
94 |
# Load ckpt & config for Video Captioning
|
95 |
+
checkpoint_path = 'checkpoints/unival_video_caption_stage_1/checkpoint_best.pt'
|
96 |
|
97 |
caption_overrides={"eval_cider":False, "beam":5, "max_len_b":22, "no_repeat_ngram_size":3, "seed":7, "unnormalized": False,
|
98 |
"bpe_dir":"utils/BPE", "video_model_path": None, "video_model_path": None, "resnet_model_path": None}
|