Fixes

app.py

@@ -33,9 +33,11 @@ def query_image(img, text_queries,max_results):
     results = processor.post_process(outputs=outputs, target_sizes=target_sizes)
     boxes, scores, labels = results[0]["boxes"], results[0]["scores"], results[0]["labels"]
     results_dict = {}
+    count = 0
     for box, score, label in zip(boxes, scores, labels):
-        results_dict[score] = {"box":box,"label":label}
-    sorted_results_dict = OrderedDict(sorted(results_dict.items(),reverse=True))
+        results_dict[count] = {"score":score.tolist(),"box":box,"label":label}
+        count += 1
+    sorted_results_dict = OrderedDict(sorted(results_dict.items(),key=lambda item: item[1]["score"],reverse=True))
 
     
     font = cv2.FONT_HERSHEY_SIMPLEX
@@ -43,7 +45,7 @@ def query_image(img, text_queries,max_results):
     score_dist = []
     count = 0
     for score in sorted_results_dict:
-        score_dist.append(round(score.tolist(),2))
+        score_dist.append(round(score,2))
         count += 1
         if (count == 10):
             break
@@ -64,7 +66,7 @@ def query_image(img, text_queries,max_results):
         else:
             y = box[3] + 25
             
-        rounded_score = round(score.tolist(),2)
+        rounded_score = round(score,2)
         img = cv2.putText(
             img, f"({rounded_score}):{text_queries[label]}", (box[0], y), font, .5, (255,0,0), 1, cv2.LINE_AA
         )