rebuild

app.py

@@ -10,7 +10,7 @@ project = hw.login(project="jayeshv")
 fs = project.get_feature_store()
 
 mr = project.get_model_registry()
-model = mr.get_model("wine_model", version=1)
+model = mr.get_model("wine_model", version=2)
 model_dir = model.download()
 
 model = joblib.load(model_dir+'/wine_model.pkl')
@@ -41,12 +41,11 @@ def wine(type,
                                  'ph', 'sulphates', 'alcohol'])
     
     print("Predicting...")
-    print(df)
     
     res = model.predict(df)
     print(res)
     
-    return int(res.round()) + 3
+    return res
 
 demo = gr.Interface(
     fn = wine,
@@ -55,17 +54,11 @@ demo = gr.Interface(
     allow_flagging = 'never',
     inputs = [
         gr.Number(value=0, label="type"),
-        gr.Number(value=0.67, label="fixed_acidity"),
-        gr.Number(value=-0.38, label="volatile_acidity"),
-        gr.Number(value=0.55, label="citric_acid"),
-        gr.Number(value=0.4, label="residual_sugar"),
-        gr.Number(value=-0.18, label="chlorides"),
-        gr.Number(value=-0.0033, label="free_sulfur_dioxide"),
-        gr.Number(value=-0.3, label="total_sulfur_dioxide"),
-        gr.Number(value=0.19, label="density"),
-        gr.Number(value=0.22, label="ph"),
-        gr.Number(value=-0.62, label="sulphates"),
-        gr.Number(value=-0.37, label="alcohol")
+        gr.Number(value=6.3, label="fixed_acidity"),
+        gr.Number(value=0.30, label="volatile_acidity"),
+        gr.Number(value=0.49, label="sulphates"),
+        gr.Number(value=9.5, label="alcohol"),
+        gr.Number(value=0.994, label="density")
     ],
     outputs="number" # output's an integer from 3-9
 )

requirements.txt

@@ -1,4 +1,4 @@
 hopsworks
 joblib
-scikit-learn==1.1.1
+scikit-learn
 xgboost