Update app.py

app.py

@@ -21,10 +21,9 @@ results = processor.post_process_object_detection(outputs, target_sizes=target_s
 persons = 0
 for score, label, box in zip(results["scores"], results["labels"], results["boxes"]):
     box = [round(i, 2) for i in box.tolist()]
-    if model.config.id2label[label.item()] == 'person':
-        print(
-            f"Detected {model.config.id2label[label.item()]} with confidence "
-            f"{round(score.item(), 3)} at location {box}"
-        )
-        $persons++
+    print(
+        f"Detected {model.config.id2label[label.item()]} with confidence "
+        f"{round(score.item(), 3)} at location {box}"
+    )
+    $persons++