Update face_to_prompt.py

face_to_prompt.py

@@ -31,7 +31,12 @@ def extract_face_prompt(img_path):
     emotion = result['dominant_emotion']
 
     # InsightFace analysis
-    model = insightface.app.FaceAnalysis(name='buffalo_l', providers=['CPUExecutionProvider'])
+    # model = insightface.app.FaceAnalysis(name='buffalo_l', providers=['CPUExecutionProvider'])
+    model = insightface.app.FaceAnalysis(
+    name='buffalo_l',
+    providers=['CPUExecutionProvider'],
+    root=os.environ.get("INSIGHTFACE_HOME", "/data/.insightface")
+)
     model.prepare(ctx_id=0)
     faces = model.get(img_np)