Spaces:
Runtime error
Runtime error
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")
|