Pre-Trained Custom Pytorch model not loading for use

Using Streamlit
1

Hey community, I was trying to load my Pytorch Deep Learning Network using joblib in streamlit.
But it’s giving me an error:
AttributeError: module ‘main’ has no attribute ‘Network’

Is it like we can not load custom nn models in streamlit?

2

Hi @Rushank_Savant, welcome to the Streamlit community!

I assume your model is a pickle file? If so, it sounds like this is your issue: