Update app.py

app.py

@@ -10,6 +10,18 @@ from sklearn.pipeline import Pipeline
 import gradio as gr
 import joblib
 
+class FastAIWrapper(BaseEstimator, RegressorMixin):
+    def __init__(self, learn):
+        self.learn = learn
+    
+    def fit(self, X, y):
+        return self
+    
+    def predict(self, X):
+        dl = self.learn.dls.test_dl(X)
+        preds, _ = self.learn.get_preds(dl=dl)
+        return preds.numpy().flatten()
+
 # Load data
 df = pd.read_csv('City_Employee_Payroll__Current__20240915.csv', low_memory=False)
 df = df.replace([np.inf, -np.inf], np.nan)