jiaqingj commited on
Commit
151b464
·
1 Parent(s): d65f94e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -64,7 +64,7 @@ def get_args():
64
  ## Models and Paths
65
  parser.add_argument("--lm_model", type=str, default='bert-base-uncased',
66
  help="Path to language model") # bert,roberta
67
- parser.add_argument("--match_model", type=str, default='clip-vit-base-patch32',
68
  help="Path to Image-Text model") # clip,align
69
  parser.add_argument("--caption_img_path", type=str, default='./examples/girl.jpg',
70
  help="file path of the image for captioning")
 
64
  ## Models and Paths
65
  parser.add_argument("--lm_model", type=str, default='bert-base-uncased',
66
  help="Path to language model") # bert,roberta
67
+ parser.add_argument("--match_model", type=str, default='openai/clip-vit-base-patch32',
68
  help="Path to Image-Text model") # clip,align
69
  parser.add_argument("--caption_img_path", type=str, default='./examples/girl.jpg',
70
  help="file path of the image for captioning")