Update app.py

app.py

@@ -49,6 +49,8 @@ def construct_transform(patch_image_size):
 tasks.register_task('caption', CaptionTask)
 tasks.register_task('refcoco', RefcocoTask)
 tasks.register_task('vqa_gen', VqaGenTask)
+tasks.register_task('snli_ve', SnliVeTask)
+
 # turn on cuda if GPU is available
 use_cuda = torch.cuda.is_available()
 # use fp16 only when GPU is available
@@ -62,6 +64,7 @@ os.system('wget https://ofa-silicon.oss-us-west-1.aliyuncs.com/checkpoints/refco
 os.system('wget https://ofa-silicon.oss-us-west-1.aliyuncs.com/checkpoints/general_demo.pt; '
           'mkdir -p checkpoints; mv general_demo.pt checkpoints/general_demo.pt')
 
+
 # Load ckpt & config for Image Captioning
 caption_overrides = {"bpe_dir": "utils/BPE", "eval_cider": False, "beam": 5,
              "max_len_b": 16, "no_repeat_ngram_size": 3, "seed": 7}