You can deploy this model and start earning money today!

weakyy/
image-captioning-baseline-model

Name: image-captioning-baseline-model
Availability: InStock

Views

No views yet

tags:

🖼️ Image Captioning Baseline Model (ResNet + LSTM + Attention)

This repository hosts the baseline model used in the Streamlit demo app that compares two captioning approaches:

To use this with our Streamlit UI:

python

encoder.load_state_dict(torch.load("encoder.pth"))
decoder.load_state_dict(torch.load("decoder.pth"))