Spaces:
Sleeping
Sleeping
Update src/models/imageCaptioning.py
Browse files
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/
|
| 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 |
}
|