Add model loading

app.py

@@ -66,6 +66,8 @@ latest_wine_string = latest_wine_df.to_string(index=False)
 # Predict the wine class
 mr = project.get_model_registry()
 model = mr.get_model("wine_class_model", version=1)
+model_dir = model.download()
+model = joblib.load(model_dir + "/wine_class_model.pkl")
 batch_data, batch_label = feature_target_split(wine_df, 'quality')
 batch_data = one_hot_encoder(batch_data, 'type')
 y_pred = model.predict(batch_data.tail(1))