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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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")