debug detection display

app.py

@@ -79,25 +79,17 @@ def get_detection(img_path):
                         (image_scale-new_size[1])//2))
     pred_dict  = model_type.end2end_detect(new_im, valid_tfms, model, class_map=class_map, detection_threshold=0.6)
 
-
-    
-    #st.write(new_im.size)
-
-  
-
-  try:
-    labels, acc = pred_dict['detection']['labels'][0], pred_dict['detection']['scores'][0]
-    acc = acc * 100
-    if labels == "Neg":
-      labels = "Negative"
-      st.success(f"Result : {labels} with {round(acc, 2)}% confidence.")
-    elif labels == "Pos":
-      labels = "Positive"
-      st.error(f"Result : {labels} with {round(acc, 2)}% confidence.")
-  except IndexError:
-    st.error("Not found ATK image! ; try to take image again..")
-    labels = "None"
-    acc = 0
+    num_count = 1
+    for i,j in zip(pred_dict['detection']['labels'], pred_dict['detection']['scores']):
+      labels = i
+      acc = round((j*100),2)
+      if labels == "Neg":
+        labels = "Negative"
+        st.success(f"{num_count}. '{labels}' with {acc} % confidence.")
+      elif labels == "Pos":
+        labels = "Positive"
+        st.error(f"{num_count}. '{labels}' with {acc} % confidence.")
+      num_count += 1
 
 def get_img_detection(img_path):
    
@@ -166,7 +158,7 @@ elif choice == "Detection (วิเคราะห์โรค)":
         st.image(get_img_detection(image))
         with st.spinner("🤖 On Working... "):
             t1 = time.perf_counter()
-            st.subheader("Process Completed!.....")
+            st.subheader("Detection result...")
             get_detection(image)
             t2 = time.perf_counter()
             st.write('time taken to run: {:.2f} sec'.format(t2-t1))