Spaces:

scikit-learn
/

tabular-playground

Running

merve HF staff commited on Sep 14, 2022

Commit

bb8750e

1 Parent(s): bed1f82

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ outputs = [gr.Dataframe(row_count = (2, "dynamic"), col_count=(1, "fixed"), labe
 def infer(inputs):
-    data = pd.DataFrame(inputs, columns=[item for item in example_dict])
     predictions = pipe.predict(inputs)
     return pd.DataFrame(predictions, columns=["results"])

 def infer(inputs):
+    data = pd.DataFrame(inputs, columns=[headers])
     predictions = pipe.predict(inputs)
     return pd.DataFrame(predictions, columns=["results"])