Hugging Face's logo

Spaces:
filipzawadka
/
wine
Sleeping

App Files Community
wine
File size: 2,051 Bytes 
55e527c
 
 
 
c414dbd
55e527c
 
 
 
81e33a0
55e527c
 
 
 
 
c51d3e5
55e527c
 
d44901c
55e527c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ebcf7a7
 
 
 
 
 
 
 
 
 
 
 
55e527c
d494e29
55e527c
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
 
import gradio as gr
import numpy as np
from PIL import Image
import requests
import os

import hopsworks
import joblib

project = hopsworks.login(api_key_value=os.environ['UNI_HOPSWORKS_API_KEY'])
fs = project.get_feature_store()

mr = project.get_model_registry()
model = mr.get_model("wine_model", version=1)
model_dir = model.download()

model = joblib.load(model_dir + "/wine_model.pkl")

os.listdir()

def wine(fixed_acidity,
    volatile_acidity,
    citric_acid,
    residual_sugar,
    chlorides,
    free_sulfur_dioxide,
    total_sulfur_dioxide,
    density,
    pH,
    sulphates,
    alcohol,
    white):
    input_list = []

    input_list.append(fixed_acidity)
    input_list.append(volatile_acidity)
    input_list.append(citric_acid)
    input_list.append(residual_sugar)
    input_list.append(chlorides)
    input_list.append(free_sulfur_dioxide)
    input_list.append(total_sulfur_dioxide)
    input_list.append(density)
    input_list.append(pH)
    input_list.append(sulphates)
    input_list.append(alcohol)
    input_list.append(white)

    res = model.predict(np.asarray(input_list).reshape(1, -1))            
    return res
        
demo = gr.Interface(
    fn=wine,
    title="Wine prediction",
    description="Wine quality prediction",
    allow_flagging="never",
    inputs=[
        gr.components.Number(1.0, label="fixed_acidity"),
        gr.components.Number(1.0, label="volatile_acidity"),
        gr.components.Number(1.0, label="citric_acid"),
        gr.components.Number(1.0, label="residual_sugar"),
        gr.components.Number(1.0, label="chlorides"),
        gr.components.Number(1.0, label="free_sulfur_dioxide"),
        gr.components.Number(1.0, label="total_sulfur_dioxide"),
        gr.components.Number(1.0, label="density"),
        gr.components.Number(1.0, label="pH"),
        gr.components.Number(1.0, label="sulphates"),
        gr.components.Number(1.0, label="alcohol"),
        gr.components.Number(1.0, label="white")
        ],
    outputs=gr.components.Number(value=-1)
    )


demo.launch()