Update app.py

app.py

@@ -121,12 +121,14 @@ labels = np.append(np.array(bc_model.dls.vocab), "Unknown")
 def predict_beetle(img):
     # Split image into smaller images of detected objects
     image_lst = detect_objects(og_image=img, model=od_model, prompt="bug . insect", device="cpu")
-        # get predictions for all segments
+    print("Objects detected")
+    # get predictions for all segments
     conf_dict_lst = []
     output_lst = []
     img_cnt = len(image_lst)
     for i in range(0,img_cnt):
         prob_ar = np.array(bc_model.predict(image_lst[i])[2])
+        print(f"Beetle classified - {i}")
         unkown_prob = unkown_prob_calc(probs=prob_ar, wedge_threshold=0.85, wedge_magnitude=5, wedge='dynamic')
         prob_ar = np.append(prob_ar, unkown_prob)
         prob_ar = np.around(prob_ar*100, decimals=1)
@@ -137,6 +139,7 @@ def predict_beetle(img):
         conf_dict = dict(sorted(conf_dict.items(), key=lambda item: item[1], reverse=True)[:top_num])
         conf_dict_lst.append(str(conf_dict)[1:-1]) # remove dictionary brackets
         result = list(zip(image_lst, conf_dict_lst))
+        print(f"Classification processed - {i}")
     return(result)