saad177 commited on
Commit
8f5efcd
1 Parent(s): cbc5f50

try with lab1 code

Browse files
Files changed (2) hide show
  1. app.py +41 -4
  2. requirements.txt +4 -0
app.py CHANGED
@@ -1,7 +1,44 @@
1
  import gradio as gr
 
 
 
2
 
3
- def greet(name):
4
- return "Hello " + name + "!!"
5
 
6
- iface = gr.Interface(fn=greet, inputs="text", outputs="text")
7
- iface.launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
+ import hopsworks
3
+ import joblib
4
+ import pandas as pd
5
 
6
+ project = hopsworks.login()
7
+ # fs = project.get_feature_store()
8
 
9
+ print("trying to dl model")
10
+ mr = project.get_model_registry()
11
+ model = mr.get_model("wine_model", version=1)
12
+ model_dir = model.download()
13
+ model = joblib.load(model_dir + "/diabetes_model.pkl")
14
+ print("Model downloaded")
15
+
16
+
17
+ def wine(age, bmi, hba1c_level, blood_glucose_level):
18
+ print("Calling wine function")
19
+ df = pd.DataFrame(
20
+ [[age, bmi, hba1c_level, blood_glucose_level]],
21
+ columns=["age", "bmi", "hba1c_level", "blood_glucose_level"],
22
+ )
23
+ print("Predicting")
24
+ print(df)
25
+ res = model.predict(df)
26
+ print(res)
27
+ return res
28
+
29
+
30
+ demo = gr.Interface(
31
+ fn=wine,
32
+ title="Diabetes Prediction",
33
+ description="Fill these values with yours",
34
+ allow_flagging="never",
35
+ inputs=[
36
+ gr.Number(label="age"),
37
+ gr.Number(label="bmi"),
38
+ gr.Number(label="hba1c_level"),
39
+ gr.Number(label="blood_glucose_level"),
40
+ ],
41
+ outputs=gr.Number(label="diabetes"),
42
+ )
43
+
44
+ demo.launch(debug=True)
requirements.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ import gradio as gr
2
+ import hopsworks
3
+ import joblib
4
+ import pandas as pd