Spaces:
Runtime error
Runtime error
File size: 2,040 Bytes
55e527c c414dbd 55e527c 66a5975 55e527c c51d3e5 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 |
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_2'])
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")
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() |