Update app.py

app.py

@@ -20,12 +20,12 @@ df.dropna(axis=0, inplace=True)
 
 
 
-inputs = [gr.Dataframe(headers = headers.keys(), row_count = (2, "dynamic"), col_count=(24,"dynamic"), label="Input Data", interactive=1)]
+inputs = [gr.Dataframe(headers = headers, row_count = (2, "dynamic"), col_count=(24,"dynamic"), label="Input Data", interactive=1)]
 outputs = [gr.Dataframe(row_count = (2, "dynamic"), col_count=(1, "fixed"), label="Predictions", headers=["Failures"])]
 
 
 def infer(inputs):
-    data = pd.DataFrame(inputs, columns=[headers])
+    data = pd.DataFrame(inputs, columns=headers)
     predictions = pipe.predict(inputs)
     return pd.DataFrame(predictions, columns=["results"])