Spaces:
Sleeping
Sleeping
Commit
·
56c09ce
1
Parent(s):
5b4afbe
Update app.py
Browse files
app.py
CHANGED
@@ -121,6 +121,7 @@ labels = np.append(np.array(bc_model.dls.vocab), "Unknown")
|
|
121 |
print("Classification model loaded")
|
122 |
|
123 |
def predict_beetle(img):
|
|
|
124 |
# Split image into smaller images of detected objects
|
125 |
image_lst = detect_objects(og_image=img, model=od_model, prompt="bug . insect", device="cpu")
|
126 |
print("Objects detected")
|
|
|
121 |
print("Classification model loaded")
|
122 |
|
123 |
def predict_beetle(img):
|
124 |
+
print("Detecting & classifying beetles...")
|
125 |
# Split image into smaller images of detected objects
|
126 |
image_lst = detect_objects(og_image=img, model=od_model, prompt="bug . insect", device="cpu")
|
127 |
print("Objects detected")
|