Fixed the functions

app.py

@@ -1,6 +1,4 @@
 import gradio as gr
-from PIL import Image
-import requests
 import hopsworks
 import joblib
 import pandas as pd
@@ -32,14 +30,18 @@ def wine(fixed_acidity, volatile_acidity, citric_acid, residual_sugar, chlorides
         'free_sulfur_dioxide': [free_sulfur_dioxide],
         'total_sulfur_dioxide': [total_sulfur_dioxide],
         'density': [density],
-        'pH': [ph],
+        'ph': [ph],
         'sulphates': [sulphates],
         'alcohol': [alcohol],
-        'type': [type],
+        'type': [type]
     })
 
+    # Change the type to integer
+    df['type'] = df['type'].map({'Red': 0, 'White': 1})
+
     # Predict the wine quality
-    return model.predict(df)[0]
+    print(f"Predicting wine quality for: {df.to_dict()}, outcome is: {model.predict(df)[0]}")
+    return round(model.predict(df)[0])
         
 demo = gr.Interface(
     fn=wine,
@@ -47,21 +49,21 @@ demo = gr.Interface(
     description="Predict the quality of a wine based on its characteristics",
     allow_flagging="never",
     inputs=[
-        gr.inputs.Slider(4, 15, 0.1, label="Fixed Acidity"),
-        gr.inputs.Slider(0, 2, 0.1, label="Volatile Acidity"),
-        gr.inputs.Slider(0, 1, 0.1, label="Citric Acid"),
-        gr.inputs.Slider(0, 20, 0.1, label="Residual Sugar"),
-        gr.inputs.Slider(0, 1, 0.1, label="Chlorides"),
-        gr.inputs.Slider(1, 100, 0.1, label="Free Sulfur Dioxide"),
-        gr.inputs.Slider(6, 300, 0.1, label="Total Sulfur Dioxide"),
-        gr.inputs.Slider(0, 1, 0.1, label="Density"),
-        gr.inputs.Slider(2, 5, 0.1, label="pH"),
-        gr.inputs.Slider(0, 1, 0.1, label="Sulphates"),
-        gr.inputs.Slider(8, 15, 0.1, label="Alcohol"),
+        gr.inputs.Slider(3.8, 15.9, 0.121, label= 'fixed_acidity'),
+        gr.inputs.Slider(0.08, 1.58, 0.015, label= 'volatile_acidity'),
+        gr.inputs.Slider(0.0, 1.66, 0.017, label= 'citric_acid'),
+        gr.inputs.Slider(0.6, 65.8, 0.652, label= 'residual_sugar'),
+        gr.inputs.Slider(0.009, 0.611, 0.006, label= 'chlorides'),
+        gr.inputs.Slider(1.0, 289.0, 2.88, label= 'free_sulfur_dioxide'),
+        gr.inputs.Slider(6.0, 440.0, 4.34, label= 'total_sulfur_dioxide'),
+        gr.inputs.Slider(0.98711, 1.03898, 0.001, label= 'density'),
+        gr.inputs.Slider(2.72, 4.01, 0.013, label= 'ph'),
+        gr.inputs.Slider(0.22, 2.0, 0.018, label= 'sulphates'),
+        gr.inputs.Slider(8.0, 14.9, 0.069, label= 'alcohol'),
         gr.inputs.Radio(["Red", "White"], label="Type"),
         ],
     outputs=[
-        gr.outputs.Label(label="Quality"),
+        gr.outputs.Textbox(label="Quality"),
         ]
     )