Update app.py

app.py

@@ -33,17 +33,17 @@ headers = [col for col in all_headers if col in df.columns]
 # Define input and output interfaces with dynamic row counts
 inputs = [gr.Dataframe(
     headers=headers,
-    #row_count=(10, "dynamic"),  # Use tuple format (min_rows, "dynamic")
+    row_count=(10, "dynamic"),  # Use tuple format (min_rows, "dynamic")
     col_count=(len(headers), "fixed"),
     label="Input Data",
     interactive=True
 )]
 
 outputs = [gr.Dataframe(
-    #row_count=(10, "dynamic"),  # Use tuple format (min_rows, "dynamic")
-    col_count=(1, "fixed"),
+    row_count=(10, "dynamic"),  # Use tuple format (min_rows, "dynamic")
+    col_count=(2, "fixed"),
     label="Predictions",
-    headers=["Depression"]
+    headers=["Name", "Depression"]
 )]
 
 
@@ -86,7 +86,7 @@ def infer(inputs):
     data = data.astype(float)
     
     predictions = pipe.predict(data)
-    return pd.DataFrame(predictions, columns=["Depression"])
+    return pd.DataFrame(predictions, columns=["Name", "Depression"])