dnth commited on
Commit
f2d04f5
·
1 Parent(s): 1bd805e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ classifier = load_learner("models/learner.pkl")
15
  classifier_labels = classifier.dls.vocab
16
 
17
  print("Loading detector")
18
- checkpoint_path = "eye_detection/models/model_checkpoint.pth"
19
  checkpoint_and_model = model_from_checkpoint(checkpoint_path)
20
  model = checkpoint_and_model["model"]
21
  model_type = checkpoint_and_model["model_type"]
 
15
  classifier_labels = classifier.dls.vocab
16
 
17
  print("Loading detector")
18
+ checkpoint_path = "models/model_checkpoint.pth"
19
  checkpoint_and_model = model_from_checkpoint(checkpoint_path)
20
  model = checkpoint_and_model["model"]
21
  model_type = checkpoint_and_model["model_type"]