Update app.py

app.py

@@ -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])
+    data = pd.DataFrame(inputs, columns=[headers])
     predictions = pipe.predict(inputs)
     return pd.DataFrame(predictions, columns=["results"])