Downloads · 30 days
0
image-caption-flickr8k
Thomas1010/image-caption-flickr8k
image-caption-flickr8k is a machine learning model from Thomas1010. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Trained model + vocabulary for an Image Caption Generator (ResNet50 features + LSTM decoder).
Access
Public
Updated Aug 29, 2026
Repo size
24.9 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt24.8 MB · 100%
At a glance
- Access
- Public
- Created
- Aug 29, 2026
- Updated
- Aug 29, 2026
- SHA
- 93f15b07
Model card
From the Hugging Face model README
Image Caption Generator — Flickr8k
Trained model + vocabulary for an Image Caption Generator (ResNet50 features + LSTM decoder).
Trained as part of Task 1.2 (AI Training '27). See the full project code and instructions on GitHub:
https://huggingface.co/Thomas1010/image-caption-flickr8k
Files
best_model.pt— trained model checkpointvocab.pkl— vocabulary used for tokenizing captions
Usage
Download both files and place them in models_store/ and data/processed/
respectively in the project repo above, then run the Streamlit app.
- Created
- Aug 29, 2026
- Updated
- Aug 29, 2026