Hugging Face's logo

Spaces:
pytholic
/
streamlit-image-classification-demo
Runtime error

App Files Community
pytholic commited on
Commit
1046bee
·
1 Parent(s): 9969b90

model path fixed

Browse files
Files changed (1) hide show
  1. app/app.py +1 -1
app/app.py CHANGED
@@ -18,7 +18,7 @@ from PIL import Image
18
  from model import Classifier
19
 
20
  # Load the model
21
- model = Classifier.load_from_checkpoint("./models/checkpoint_old.ckpt")
22
  model.eval()
23
 
24
  # Define labels
 
18
  from model import Classifier
19
 
20
  # Load the model
21
+ model = Classifier.load_from_checkpoint("./models/checkpoint.ckpt")
22
  model.eval()
23
 
24
  # Define labels