Not able to deploy tensorflow model

Hello All,

The github repo can be accessed from here: repo.

when I try to deploy it on streamlit, I get this error message

Can anyone please help me out.

Check out the cuda and cudnn versions that are compatible with Tensorflow.

Hello WhySecond,

I figured out a way to solve that Issue.
The problem was caused by TextVectorization layer.

However, now that I’,m using universal sentence encoder, the model size has become 1.1 GB.
So, I’m not able to git push it.

Will modifying the setup.sh script to download the model, unzipping it work…?

Thanks in advance.

1 Like

Check out the cuda and cudnn versions that are compatible with Tensorflow.

Also, can you please point me to where this is available…?

Thanks

1 Like

mkdir deploy_models
wget pubmed-model.zip - Google Drive
unzip pubmed-model.zip -d deploy_models

I added these lines to setup.sh.

Please correct me if I’m wrong.

I don’t really know either.

Do you have any saved models added to the repo?

No worries,

I deployed the model using a docker container and managed it to get working.

Also the issue was with using TextVectorization layer and a callback function.

Thanks,