Update app.py

app.py

@@ -25,8 +25,27 @@ headers = [col for col in all_headers if col in df.columns]
 
 # Define input and output interfaces
 #inputs = [gr.Dataframe(headers=headers, row_count=(2, "dynamic"), col_count=(len(headers), "fixed"), label="Input Data", interactive=True)]
-inputs = [gr.Dataframe(headers=all_headers, row_count=(2, "dynamic"), col_count=(len(all_headers), "fixed"), label="Input Data", interactive=True)]
-outputs = [gr.Dataframe(row_count=(2, "dynamic"), col_count=(1, "fixed"), label="Predictions", headers=["Depression"])]
+
+#working code that returns only 2 rows in output
+#inputs = [gr.Dataframe(headers=all_headers, row_count=(2, "dynamic"), col_count=(len(all_headers), "fixed"), label="Input Data", interactive=True)]
+#outputs = [gr.Dataframe(row_count=(2, "dynamic"), col_count=(1, "fixed"), label="Predictions", headers=["Depression"])]
+
+# Define input and output interfaces with dynamic row counts
+inputs = [gr.Dataframe(
+    headers=headers,
+    row_count="dynamic",  # Remove the tuple and just use "dynamic"
+    col_count=(len(headers), "fixed"),
+    label="Input Data",
+    interactive=True
+)]
+
+outputs = [gr.Dataframe(
+    row_count="dynamic",  # Remove the tuple and just use "dynamic"
+    col_count=(1, "fixed"),
+    label="Predictions",
+    headers=["Depression"]
+)]
+
 
 #def infer(inputs):
     #data = pd.DataFrame(inputs, columns=headers)