Update face_to_prompt.py

face_to_prompt.py

@@ -12,10 +12,16 @@ def extract_face_prompt(img_path):
     deepface_result = DeepFace.analyze(
         img_path, actions=['age', 'gender', 'race', 'emotion'], enforce_detection=False
     )
-    age = deepface_result['age']
-    gender = deepface_result['dominant_gender']
-    race = deepface_result['dominant_race']
-    emotion = deepface_result['dominant_emotion']
+
+    if isinstance(deepface_result, list):
+        result = deepface_result[0]
+    else:
+        result = deepface_result
+        
+    age = result['age']
+    gender = result['dominant_gender']
+    race = result['dominant_race']
+    emotion = result['dominant_emotion']
 
     # InsightFace analysis
     model = insightface.app.FaceAnalysis(name='buffalo_l', providers=['CPUExecutionProvider'])