Spaces:

azamat
/

emoface

Runtime error

azamat commited on Sep 26, 2022

Commit

272de03

•

1 Parent(s): 0bb1252

Fix bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,12 +3,12 @@ import pandas as pd
 from deepface import DeepFace
 def analyze(img_path):
-    face_analysis = DeepFace.analyze(img_path = img_path)
     return pd.DataFrame({
-        'gender': face_analysis['gender'],
-        'age': face_analysis['age'],
-        'dominant_emotion': face_analysis['dominant_emotion'],
-        'dominant_race': face_analysis['dominant_race']
     })
 def main():

 from deepface import DeepFace
 def analyze(img_path):
+    face_analysis = DeepFace.analyze(img_path = img_path, enforce_detection=False)
     return pd.DataFrame({
+        'gender': face_analysis.get('gender', None),
+        'age': face_analysis.get('age', None),
+        'dominant_emotion': face_analysis.get('dominant_emotion', None),
+        'dominant_race': face_analysis.get('dominant_race', None)
     })
 def main():