mgbam commited on
Commit
5121009
·
verified ·
1 Parent(s): a1e0218

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +1 -1
config.py CHANGED
@@ -9,7 +9,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
9
  OPENAI_DEFAULT_MODEL = "gpt-4o"
10
 
11
  # Hugging Face Model for Medical Image Analysis
12
- HF_IMAGE_MODEL = "Salesforce/blip-image-captioning-base"
13
 
14
  # PubMed and summarization settings
15
  MAX_PUBMED_RESULTS = 5
 
9
  OPENAI_DEFAULT_MODEL = "gpt-4o"
10
 
11
  # Hugging Face Model for Medical Image Analysis
12
+ HF_IMAGE_MODEL = "nlpconnect/vit-gpt2-image-captioning"
13
 
14
  # PubMed and summarization settings
15
  MAX_PUBMED_RESULTS = 5