Update app.py

app.py

@@ -18,8 +18,6 @@ df = datasets.load_dataset("silvaKenpachi/mental_health")
 df = df["train"].to_pandas()
 df.dropna(axis=0, inplace=True)
 
-feature_columns = [col for col in df.columns if col != 'Premium Amount']
-
 
 
 inputs = [gr.Dataframe(headers = headers, row_count = (2, "dynamic"), col_count=(len(headers), "dynamic"), label="Input Data", interactive=1)]