Updated app at sön 19 nov 2023 22:08:17 CET

app.py

@@ -5,7 +5,14 @@ import hopsworks
 import joblib
 import pandas as pd
 
-
+# Settings
+# - - Names
+# - - - Models
+model_wine_dir = "wine_models"
+model_red_wine_name = "wine_red_model"
+model_red_wine_version = 1
+model_white_wine_name = "wine_white_model"
+model_white_wine_version = 1
 
 def wine_quality(wine_type, fixed_acidity, volatile_acidity, citric_acid, residual_sugar, chlorides,
                  free_sulfur_dioxide, total_sulfur_dioxide, density, ph, sulphates, alcohol):
@@ -18,16 +25,44 @@ def wine_quality(wine_type, fixed_acidity, volatile_acidity, citric_acid, residu
     print("Predicting...")
     print(df)
 
-
-
-    if wine_type == 'Red':
-        quality = "Red Wine Quality Prediction Placeholder"
+    if wine_type.lower() == 'red':
+        quality = red_wine_local_model.predict(df)
+    elif wine_type.lower() == 'white':
+        quality = white_wine_local_model.predict(df)
     else:
-        quality = "White Wine Quality Prediction Placeholder"
+        quality = "Unknown wine type. Only 'red' and 'white' are known."
 
     return quality
 
 
+print("Logging in to Hopsworks...")
+project = hopsworks.login()
+
+print("Getting feature store...")
+fs = project.get_feature_store()
+
+print("Getting model registry...")
+mr = project.get_model_registry()
+
+print("Getting model for red wine: ...")
+model_red_wine = mr.get_model(model_red_wine_name, version=model_red_wine_version)
+
+print("Downloading model for red wine...")
+model_red_dir = model_red_wine.download()
+
+print("Initializing model red wine locally...")
+red_wine_local_model = joblib.load(model_red_dir + f"/{model_red_wine_name}.pkl")
+
+print("Getting model for white wine: ...")
+model_white_wine = mr.get_model(model_white_wine_name, version=model_white_wine_version)
+
+print("Downloading model for white wine...")
+model_white_dir = model_white_wine.download()
+
+print("Initializing model white wine locally...")
+white_wine_local_model = joblib.load(model_white_dir + f"/{model_white_wine_name}.pkl")
+
+
 # Define the Gradio interface
 iface = gr.Interface(fn=wine_quality,
                      inputs=["text", "number", "number", "number", "number", "number",