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