Update app.py

app.py

@@ -79,17 +79,20 @@ 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)
 
-    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
+    if (len(pred_dict['detection']['labels'])) > 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
+    else: # when not found imgs
+      st.warning("Not found 'Antigen Test Kit' image ; please recheck and try again !!")
 
 def get_img_detection(img_path):