Spaces:
Sleeping
Sleeping
Commit
·
5b4afbe
1
Parent(s):
c3d43c5
Update app.py
Browse files
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")
|