Merge pull request #4 from GrigoriiTarasov/dev

Fastfixes
GrigoriiTarasov · Jun 16, 2024 · 77951cc · 77951cc
2 parents cc2cf54 + 2e9eddd
commit 77951cc
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 397 deletions.
diff --git a/age_module/app/app.py b/age_module/app/app.py
@@ -32,7 +32,7 @@ def predict_one_frame(frame: np.array, predictor: PipelinePredictor) -> List[Dic
     client_output = []
     for face_obj, age_pred in zip(extraction_obj, age_preds):
         # Assuming age_pred is a numpy array with one element, extract that element
-        age = age_pred
+        age = age_pred.item()
         client_output.append(AdaptPredAPI(face_obj, age).client_output())
 
     return client_output