YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
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.
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support