Update app.py

app.py

@@ -2,6 +2,7 @@ import sklearn
 import gradio as gr
 import joblib
 import pandas as pd
+import datasets
 
 pipe = joblib.load("./model.pkl")
 
@@ -13,7 +14,9 @@ with open("./config.json") as f:
     config_dict = eval(f.read())
 headers = config_dict["sklearn"]["columns"]
 
-example_dict = config_dict["sklearn"]["example_input"]
+df = datasets.load_dataset("merve/supersoaker-failures")
+df = df["train"].to_pandas()
+df.dropna(axis=0, inplace=True)
 df = pd.DataFrame.from_dict(example_dict,orient='index').transpose()
 
 
@@ -28,4 +31,4 @@ def infer(inputs):
     return pd.DataFrame(predictions, columns=["results"])
 
 gr.Interface(infer, inputs = inputs, outputs = outputs, title = title,
-            description = description, examples=df.tail(3), cache_examples=False).launch(debug=True)
+            description = description, examples=df.head(3), cache_examples=False).launch(debug=True)