Update app.py

app.py

@@ -38,6 +38,7 @@ def main():
         # st.markdown("---")  # Add a separator line
         st.header("Settings")
         
+        num_lines = st.text_input("% of lines to be processed:", value=100)
         graph_select = st.checkbox("Show Graphics", value= True)
         feat_imp_select = st.checkbox("Feature Importance", value= False)
 
@@ -96,7 +97,7 @@ def main():
             else:
                 insurance_claims = pd.read_csv(selected_csv)
 
-            insurance_claims = insurance_claims.head(1000)
+            insurance_claims = insurance_claims.head(insurance_claims.shape[0]*num_lines/100)
             
             st.header("Inference Description")
             insurance_claims.describe().T
@@ -228,7 +229,7 @@ def main():
             else:
                 insurance_claims = pd.read_csv(selected_csv)
 
-            insurance_claims = insurance_claims.head(1000)
+            insurance_claims = insurance_claims.head(insurance_claims.shape[0]*num_lines/100)
             
             all_columns = insurance_claims.columns.tolist()
             selected_columns = st.multiselect("Choose columns", all_columns, default=all_columns)