Yes you can decouple the streamlit and use functions in collab, you'll have to save the models in drive then and make changes in path. I will push the jupyter notebook I had for this experiment in the repo.