Update app.py
Browse files
app.py
CHANGED
|
@@ -175,8 +175,8 @@ move2gpu(audio_caption_models, general_cfg)
|
|
| 175 |
caption_generator = caption_task.build_generator(caption_models, caption_cfg.generation)
|
| 176 |
refcoco_generator = refcoco_task.build_generator(refcoco_models, refcoco_cfg.generation)
|
| 177 |
vqa_generator = vqa_task.build_generator(vqa_models, vqa_cfg.generation)
|
| 178 |
-
vqa_generator.zero_shot = True
|
| 179 |
-
vqa_generator.constraint_trie = None
|
| 180 |
general_generator = general_task.build_generator(general_models, general_cfg.generation)
|
| 181 |
|
| 182 |
video_caption_generator = caption_task.build_generator(video_caption_models, video_caption_cfg.generation)
|
|
|
|
| 175 |
caption_generator = caption_task.build_generator(caption_models, caption_cfg.generation)
|
| 176 |
refcoco_generator = refcoco_task.build_generator(refcoco_models, refcoco_cfg.generation)
|
| 177 |
vqa_generator = vqa_task.build_generator(vqa_models, vqa_cfg.generation)
|
| 178 |
+
# vqa_generator.zero_shot = True
|
| 179 |
+
# vqa_generator.constraint_trie = None
|
| 180 |
general_generator = general_task.build_generator(general_models, general_cfg.generation)
|
| 181 |
|
| 182 |
video_caption_generator = caption_task.build_generator(video_caption_models, video_caption_cfg.generation)
|