Update app.py

app.py

@@ -97,9 +97,9 @@ def main():
             else:
                 insurance_claims = pd.read_csv(selected_csv)
 
-            num_rows_2 = int(insurance_claims.shape[0]*int(num_lines)/100)
-            #insurance_claims = insurance_claims.head(100)
-            st.write("Rows to be processed: " + str(num_rows_2))
+            num_rows = int(insurance_claims.shape[0]*int(num_lines)/100)
+            insurance_claims = insurance_claims.head(num_rows)
+            st.write("Rows to be processed: " + str(num_rows))
             
             st.header("Inference Description")
             insurance_claims.describe().T
@@ -231,9 +231,9 @@ def main():
             else:
                 insurance_claims = pd.read_csv(selected_csv)
 
-            #num_rows = int(insurance_claims.shape[0]*num_lines/100)
-            #insurance_claims = insurance_claims.head(num_rows)
-            #st.write("Rows to be processed: " + str(num_rows))
+            num_rows = int(insurance_claims.shape[0]*int(num_lines)/100)
+            insurance_claims = insurance_claims.head(num_rows)
+            st.write("Rows to be processed: " + str(num_rows))
             
             all_columns = insurance_claims.columns.tolist()
             selected_columns = st.multiselect("Choose columns", all_columns, default=all_columns)