Not able to deploy tensorflow model

[shreyas0906]

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.

[WhySecond]

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

[shreyas0906]

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.

[shreyas0906]

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

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

Thanks

[shreyas0906]

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.

[WhySecond]

I don’t really know either.

Do you have any saved models added to the repo?

[shreyas0906]

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,