mgbam commited on
Commit
a7b0517
·
verified ·
1 Parent(s): 4d04338

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +1 -1
config.py CHANGED
@@ -14,4 +14,4 @@ DEFAULT_CHUNK_SIZE = 512
14
 
15
  # A stable image captioning model recognized by the HF "image-to-text" pipeline
16
  # Example: "nlpconnect/vit-gpt2-image-captioning" or "Salesforce/blip-image-captioning-base"
17
- IMAGE_MODEL_NAME = "nlpconnect/vit-gpt2-image-captioning"
 
14
 
15
  # A stable image captioning model recognized by the HF "image-to-text" pipeline
16
  # Example: "nlpconnect/vit-gpt2-image-captioning" or "Salesforce/blip-image-captioning-base"
17
+ IMAGE_MODEL_NAME = "Salesforce/blip-image-captioning-base"