Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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")
|