| from typing import Dict, List, Union | |
| import os | |
| import joblib | |
| import numpy as np | |
| features = ["fixed_acidity", "volatile_acidity", "citric_acid", "residual_sugar", "chlorides", "free_sulfur_dioxide", "total_sulfur_dioxide", "density", "pH", "sulphates", "alcohol"] | |
| model = joblib.load("sklearn_model.joblib") | |
| model_path = "model.pkl" | |
| with open(model_path, mode="bw") as f: | |
| pickle.dump(model, file=f) | |