Datatrooper/wine

1import pandas as pd2from joblib import dump, load3import gradio as gr4 5rfc = load('wine_pred.joblib') 6 7def predict_quality(fixed_acidity, volatile_acidity, citric_acid, residual_sugar, chlorides, free_sulfur_dioxide, total_sulfur_dioxide, density, ph, sulphates, alcohol):8    df = pd.DataFrame.from_dict(9        {10            'fixed acidity': fixed_acidity,11            'volatile acidity': volatile_acidity,12            'citric acid': citric_acid,13            'residual sugar': residual_sugar,14            'chlorides': chlorides,15            'free sulfur dioxide': free_sulfur_dioxide,16            'total sulfur dioxide': total_sulfur_dioxide,17            'density': density,18            'pH': ph,19            'sulphates': sulphates,20            'alcohol': alcohol21        }, orient='index').T22        23    return rfc.predict(df)[0]24 25 26 27iface = gr.Interface(28    predict_quality,29    [30        gr.inputs.Slider(0, 20),31        gr.inputs.Slider(0, 2),32        gr.inputs.Slider(0, 1),33        gr.inputs.Slider(0, 20),34        gr.inputs.Slider(0, 1),35        gr.inputs.Slider(0, 100),36        gr.inputs.Slider(0, 300),37        gr.inputs.Slider(0, 2),38        gr.inputs.Slider(0, 5),39        gr.inputs.Slider(0, 2),40        gr.inputs.Slider(0, 15)41 42    ],43    "label",44    examples=[45        [7.4, 0.7, 0.0, 1.9, 0.076, 11.0, 34.0, 0.9978, 3.51, 0.56, 9.4],46    ],47    interpretation="default",48    title="Wine quality regressor",49    description="Predict wine quality based on properties"50)51 52if __name__ == "__main__":53    iface.launch()