Update app.py

app.py

@@ -47,8 +47,8 @@ ages_genders_detector = HumanAG_ONNX(
 
 def detect_objects(img, thr):
     # infer img
-    print(img)
-    img = cv2.imread(img)
+    if isinstance(img, str):
+        img = cv2.imread(img)
     objs = person_detector.infer(img, threshold=thr)
     # logger.info(f"Number of objs: {len(objs)}")
 
@@ -65,6 +65,8 @@ def detect_objects(img, thr):
 
 def detect_poses(img, thr):
     # infer img
+    if isinstance(img, str):
+        img = cv2.imread(img)
     objs = person_detector.infer(img, threshold=thr)
     objs = [obj for obj in objs if obj['confidence'] > 0.5]
     output_text = ""
@@ -102,6 +104,8 @@ def detect_poses(img, thr):
 
 
 def detect_ages_gender(img, thr):
+    if isinstance(img, str):
+        img = cv2.imread(img)
     # infer img
     objs = person_detector.infer(img, threshold=thr)
     objs = [obj for obj in objs if obj['confidence'] > 0.5]