Update app.py

app.py

@@ -109,7 +109,7 @@ def extract_heads(filepath, frame0):
         array_from_file = np.delete(array_from_file,np.s_[7:10], axis=1)
         nbperson = np.unique(array_from_file[:,1]).shape[0]
         st.subheader('Display some detected heads :')
-        st.write(' Number of heads detected : ', nbperson )
+        st.write(' Number of detected heads : ', nbperson )
 
         rows = 5
         cols = 10
@@ -125,10 +125,13 @@ def extract_heads(filepath, frame0):
 
 #*********************************************************************************************
 def display_heads_(nbperson, listhead):
+    rows = 10
+    cols = 5
     k = 0
-    for i in range(1, 10):
-        cols = st.columns(5)
-        for j in range(1, 5):
+    for i in range(1, rows):
+        #cols = st.columns(cols )
+        cols = st.beta_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)