Update app.py

app.py

@@ -1,9 +1,22 @@
 import pandas as pd
 import numpy as np
 from sklearn.ensemble import VotingRegressor
+from sklearn.base import BaseEstimator, RegressorMixin
 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 your data and trained model
 df = pd.read_csv('City_Employee_Payroll__Current__20240915.csv', low_memory=False)
 ensemble = joblib.load('ensemble_model.joblib')