mshukor HF Staff commited on
Commit
a8280cf
·
1 Parent(s): 2c7f33c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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)