Update app.py

app.py

@@ -125,19 +125,22 @@ def extract_heads(filepath, frame0):
 
 #*********************************************************************************************
 def display_heads_(nbperson, listhead):
-    rows = 10
-    cols = 5
+    rows_ = 10
+    cols_ = 5
     k = 0
-    for i in range(1, rows):
-        cols = st.columns(cols)
-        for j in range(1, cols):
+    for i in range(1, rows_):
+        cols = st.columns(cols_)
+        for j in range(1, cols_):
             k = k + 1
             cols[j].header("Person " + str(k))
-            cols[j].image(listhead[k], use_column_width=True, width=150)
+            cols[j].image(listhead[k], use_column_width=True, width=150, caption = str(k) )
     return 
 
 #*********************************************************************************************
 
+
+
+
  
 
 
@@ -156,6 +159,13 @@ def main():
         st.success("Click again to retry or try a different video by uploading")
         nbperson, listhead = extract_heads(filepath, frame0) 
         display_heads_(nbperson, listhead)
+        if st.button('Continue to face analysis!'):
+            os.system('pip install deepface')
+            from deepface import DeepFace
+            obj = DeepFace.analyze(img_path = listhead[1], actions = ['age', 'gender', 'race', 'emotion'])
+            st.write(obj)
+
+            
    
     return