Detectron2 based NN on Streamlit

Hi.
I am attempting to use Streamlit for my Detectron2 based neural network, however I cant seem to be able to install torch to streamlit. Its in the requirements, but it is clearly not installed correctly. I have tried specifying version, even as a git commit, but no luck.

Any thoughts on how to fix this?
Used git (nearly blank Det2 NN, used as an example): github page
Thanks!

As long as you add it to your requirements file as “torch” (not pytorch – I’ve had my app crash when I’ve tried importing it as pytorch or listing it as pytorch in my requirements file), it should work. Here’s a repo for a Streamlit app that imports torch that I’ve been able to deploy succesfully.

This topic was automatically closed 365 days after the last reply. New replies are no longer allowed.