Streamlit error

[Adwait-0608-Bhosale]

Hello @mrdbourke sir,

I loaded my save model for my image classification project on pycharm, for building my Streamlit web application but i am getting the below error after loading my model.

('Keyword argument not understood:', 'keepdims')

[mrdbourke]

Hey @Adwait-0608-Bhosale,

This will depend on how you're loading your model.

I'd check your code to see if you use the "keepdims" keyword somewhere and go back from there.

How are you loading the model into Streamlit?