Search code examples
pythonmachine-learningdeploymentstreamlit

A machine learning model that classifies dogs


#question Hello. There was a problem deploying to streamlit. I did the same as in the video but it says something wrong.

I wrote the code correctly, I used venv. but for some reason it is not working. enter image description here

text

My GitHub profile shows the problem in full.


Solution

  • Run the code in kaggle or google colab notebooks. Then run this !nvidia-smi

    find you have cuda, it shows cuda version. and use device = torch.device("cpu").

    When you training the model use model_0 = modelname().to("cuda")