Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -101,7 +101,7 @@ from huggingface_hub import hf_hub_download
|
|
| 101 |
from src.inference import load_attention_model, caption_image
|
| 102 |
from src.utils import get_device
|
| 103 |
|
| 104 |
-
repo_id = "OmarGamal48812/
|
| 105 |
ckpt_path = hf_hub_download(repo_id=repo_id, filename="attention_gru_glove.pth")
|
| 106 |
vocab_path = hf_hub_download(repo_id=repo_id, filename="vocab.pkl")
|
| 107 |
|
|
|
|
| 101 |
from src.inference import load_attention_model, caption_image
|
| 102 |
from src.utils import get_device
|
| 103 |
|
| 104 |
+
repo_id = "OmarGamal48812/flickr-captioning"
|
| 105 |
ckpt_path = hf_hub_download(repo_id=repo_id, filename="attention_gru_glove.pth")
|
| 106 |
vocab_path = hf_hub_download(repo_id=repo_id, filename="vocab.pkl")
|
| 107 |
|