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}
|