wine app
Browse files- README.md +1 -1
- app.py +60 -0
- requirements.txt +3 -0
README.md
CHANGED
@@ -4,7 +4,7 @@ emoji: π
|
|
4 |
colorFrom: purple
|
5 |
colorTo: gray
|
6 |
sdk: gradio
|
7 |
-
sdk_version:
|
8 |
app_file: app.py
|
9 |
pinned: false
|
10 |
---
|
|
|
4 |
colorFrom: purple
|
5 |
colorTo: gray
|
6 |
sdk: gradio
|
7 |
+
sdk_version: 3.14.0
|
8 |
app_file: app.py
|
9 |
pinned: false
|
10 |
---
|
app.py
ADDED
@@ -0,0 +1,60 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import gradio as gr
|
2 |
+
from PIL import Image
|
3 |
+
import requests
|
4 |
+
import hopsworks
|
5 |
+
import joblib
|
6 |
+
import pandas as pd
|
7 |
+
|
8 |
+
project = hopsworks.login()
|
9 |
+
fs = project.get_feature_store()
|
10 |
+
|
11 |
+
|
12 |
+
mr = project.get_model_registry()
|
13 |
+
model = mr.get_model("wine_model", version=2)
|
14 |
+
model_dir = model.download()
|
15 |
+
model = joblib.load(model_dir + "/wine_model.pkl")
|
16 |
+
print("Model downloaded")
|
17 |
+
|
18 |
+
def wine(type,fixed_acidity,volatile_acidity,citric_acid,residual_sugar,chlorides,free_sulfur_dioxide,total_sulfur_dioxide,density,ph,sulphates,alcohol):
|
19 |
+
print("Calling function")
|
20 |
+
# df = pd.DataFrame([[sepal_length],[sepal_width],[petal_length],[petal_width]],
|
21 |
+
df = pd.DataFrame([[type,fixed_acidity,volatile_acidity,citric_acid,residual_sugar,chlorides,free_sulfur_dioxide,total_sulfur_dioxide,density,ph,sulphates,alcohol]],
|
22 |
+
columns=["type","fixed_acidity","volatile_acidity","citric_acid","residual_sugar","chlorides","free_sulfur_dioxide","total_sulfur_dioxide","density","ph","sulphates","alcohol"])
|
23 |
+
print("Predicting")
|
24 |
+
print(df)
|
25 |
+
# 'res' is a list of predictions returned as the label.
|
26 |
+
res = model.predict(df)
|
27 |
+
# We add '[0]' to the result of the transformed 'res', because 'res' is a list, and we only want
|
28 |
+
# the first element.
|
29 |
+
# print("Res: {0}").format(res)
|
30 |
+
print(res)
|
31 |
+
if (res==1):
|
32 |
+
wine_url = "https://media.istockphoto.com/id/117068556/sv/foto/bad-wine.jpg?s=2048x2048&w=is&k=20&c=wLOisv5qh9N8bp8AISRo1yP2nOjq_ouvt4sWeZ11yy0="
|
33 |
+
else :
|
34 |
+
wine_url = "https://i.ytimg.com/vi/9wFm7wTJ7JU/maxresdefault.jpg"
|
35 |
+
# wine_url = "https://raw.githubusercontent.com/featurestoreorg/serverless-ml-course/main/src/01-module/assets/" + res[0] + ".png"
|
36 |
+
img = Image.open(requests.get(wine_url, stream=True).raw)
|
37 |
+
return img
|
38 |
+
|
39 |
+
demo = gr.Interface(
|
40 |
+
fn=wine,
|
41 |
+
title="Wine quality Predictive Analytics",
|
42 |
+
description="Experiment with some factors to predict what quality it is.",
|
43 |
+
allow_flagging="never",
|
44 |
+
inputs=[
|
45 |
+
gr.inputs.Number(default=1.0, label="type"),
|
46 |
+
gr.inputs.Number(default=7.2, label="fixed_acidity"),
|
47 |
+
gr.inputs.Number(default=0.33, label="volatile_acidity"),
|
48 |
+
gr.inputs.Number(default=0.31, label="citric_acid"),
|
49 |
+
gr.inputs.Number(default=5.44, label="residual_sugar"),
|
50 |
+
gr.inputs.Number(default=0.056, label="chlorides"),
|
51 |
+
gr.inputs.Number(default=30.53, label="free_sulfur_dioxide"),
|
52 |
+
gr.inputs.Number(default=115.74, label="total_sulfur_dioxide"),
|
53 |
+
gr.inputs.Number(default=0.995, label="density"),
|
54 |
+
gr.inputs.Number(default=3.21, label="ph"),
|
55 |
+
gr.inputs.Number(default=0.53, label="sulphates"),
|
56 |
+
gr.inputs.Number(default=10.49, label="alcohol"),
|
57 |
+
],
|
58 |
+
outputs=gr.Image(type="pil"))
|
59 |
+
|
60 |
+
demo.launch(debug=True)
|
requirements.txt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
hopsworks
|
2 |
+
joblib
|
3 |
+
scikit-learn==1.1.1
|