fix bug 4

app.py

@@ -68,11 +68,13 @@ def process_inference_result(result, class_colors):
 
   return img, label_counts
 
-def get_detection_count_display(classes):
-  if(len(classes)): 
+def get_detection_count_display(classes): 
+  class_names = classes["class"]
+  counts = classes["count"]
+  if(len(classes)):
     disp_str = "[Aircraft detected] "
-    for class_name, count in classes.items():
-      disp_str += class_name + ": " + str(count) + "  " 
+    for i, c in enumerate(class_names):
+      disp_str += c + ": " + str(counts[i]) + "  " 
   else:
     disp_str = "[No aircraft detected]"
   return disp_str