Update wine_model_final.py