Uzaiir commited on
Commit
4af5863
·
verified ·
1 Parent(s): c073b36

Update src/models/imageCaptioning.py

Browse files
Files changed (1) hide show
  1. src/models/imageCaptioning.py +1 -1
src/models/imageCaptioning.py CHANGED
@@ -6,7 +6,7 @@ import os
6
 
7
  hf_token = os.environ.get("HUGGINGFACE_API_TOKEN")
8
 
9
- API_URL = "https://api-inference.huggingface.co/models/Salesforce/blip-image-captioning-base" # Or any other captioning model
10
  headers = {
11
  "Authorization": f"Bearer {hf_token}"
12
  }
 
6
 
7
  hf_token = os.environ.get("HUGGINGFACE_API_TOKEN")
8
 
9
+ API_URL = "https://api-inference.huggingface.co/models/nlpconnect/vit-gpt2-image-captioning"
10
  headers = {
11
  "Authorization": f"Bearer {hf_token}"
12
  }