ChristopherMarais commited on
Commit
5b4afbe
·
1 Parent(s): c3d43c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -114,7 +114,7 @@ def detect_objects(og_image, model=od_model, prompt="bug . insect", device="cpu"
114
 
115
 
116
  # load beetle classifier model
117
- repo_id="ChristopherMarais/beetle-model"
118
  bc_model = from_pretrained_fastai(repo_id)
119
  # get class names
120
  labels = np.append(np.array(bc_model.dls.vocab), "Unknown")
 
114
 
115
 
116
  # load beetle classifier model
117
+ repo_id="ChristopherMarais/beetle-model-mini"
118
  bc_model = from_pretrained_fastai(repo_id)
119
  # get class names
120
  labels = np.append(np.array(bc_model.dls.vocab), "Unknown")