Update app.py

app.py

@@ -87,11 +87,16 @@ def detect_objects(model_name,url_input,image_input,threshold):
     # return [viz_img, processed_outputs]
     # print(type(viz_img))
 
-    final_str = ""
+    final_str_abv = ""
+    final_str_else = ""
     for score, label, box in zip(processed_outputs["scores"], processed_outputs["labels"], processed_outputs["boxes"]):
         box = [round(i, 2) for i in box.tolist()]
-        final_str += f"Detected `{model.config.id2label[label.item()]}` with confidence `{score.item(), 3}` at location `{box}`\n"
+        if score.item() >= threshold:
+            final_str_abv += f"Detected `{model.config.id2label[label.item()]}` with confidence `{round(score.item(), 3)}` at location `{box}`\n"
+        else:
+            final_str_else += f"Detected `{model.config.id2label[label.item()]}` with confidence `{round(score.item(), 3)}` at location `{box}`\n"
 
+    final_str = "{ABOVE THRESHOLD OR EQUAL:*^50}\n".format("centered") + final_str_abv + "\n{BELOW THRESHOLD:*^50}\n".format("centered")+final_str_else
         
     return viz_img, final_str