Train the model on colab and use it in local

Using Streamlit
1

Is it possible to train my model and colab and create streamlit application on local PC!?

2

Yes…train the model, store it as pickle file and then use it (import) for inference - create streamlit application on local pc…

3

Thanks a lot!, since I am a beginner, can you share some examples!

4

@seanbenhur49 Store your pickle file (*.pkl) in a directory where you place streamlit python file…
#open, load and use it like below…its just an example…
import pickle

pickle_file = open(".pkl",“rb”)
m = pickle.load(pickle_file)
m.predict(**)

5

Thanks a lot for the explanation!!

6

Hey, I’m using same procedure to load my pytorch deep learning model. But it is giving me following error:
AttributeError: Can’t get attribute ‘Network’ on <module ‘main’ from ‘E:\Deployment Projects\FeCare\App.py’>

Can u suggest something?

7

Please, in my on case.
It display

ModuleNotFoundError: no module named’sklearn’

Thank you for help in advance.

8

Hi there, I have this example that could be of some help for starters…

https://github.com/napoles-uach/streamlit_apps/blob/main/Streamlit_Colab/07_Streamlit__Colab_DogCat.ipynb

Regards!!