Upload inference.py with huggingface_hub
Browse files- inference.py +3 -0
inference.py
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import joblib, pandas as pd
|
| 2 |
+
model = joblib.load('best_model.joblib')
|
| 3 |
+
def predict(df): return model.predict(df)
|