Update app.py

app.py

@@ -130,10 +130,12 @@ def predict_beetle(img):
         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)
-        
+        # only show the top 5 predictions
+        # Sorting the dictionary by value in descending order and taking the top items
+        top_num = 3
         conf_dict = {labels[i]: float(prob_ar[i]) for i in range(len(prob_ar))}
-        conf_dict = dict(sorted(conf_dict.items(), key=lambda item: item[1], reverse=True))
-        conf_dict_lst.append(str(conf_dict))
+        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))
     return(result)