Spaces:
Running
Running
Thanabodin
commited on
Commit
•
d09049f
1
Parent(s):
d3872f1
test model and upload model pipeline
Browse files- app.py +26 -4
- filename.pkl +0 -0
- scaler.pkl +0 -0
app.py
CHANGED
@@ -1,7 +1,29 @@
|
|
1 |
import gradio as gr
|
|
|
|
|
|
|
2 |
|
3 |
-
def
|
4 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
5 |
|
6 |
-
|
7 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
import gradio as gr
|
2 |
+
import joblib
|
3 |
+
import numpy as np
|
4 |
+
from sklearn.preprocessing import StandardScaler
|
5 |
|
6 |
+
def make_prediction(MedInc,HouseAge,AveRooms,AveBedrms,Population,AveOccup,Latitude,Longitude):
|
7 |
+
scaler = joblib.load('scaler.pkl')
|
8 |
+
model = joblib.load("filename.pkl")
|
9 |
+
new_data = np.array([[MedInc,HouseAge,AveRooms,AveBedrms,Population,AveOccup,Latitude,Longitude]])
|
10 |
+
scaler_new = scaler.transform(new_data)
|
11 |
+
preds = model.predict(scaler_new)
|
12 |
+
pred = preds[0, 0]
|
13 |
+
return f"{pred:.2f}"
|
14 |
|
15 |
+
#Create the input component for Gradio since we are expecting 4 inputs
|
16 |
+
MedInc = gr.Number(label = "Enter MedInc")
|
17 |
+
HouseAge = gr.Number(label = "Enter HouseAge")
|
18 |
+
AveRooms = gr.Number(label = "Enter AveRooms")
|
19 |
+
AveBedrms = gr.Number(label = "Enter AveBedrms")
|
20 |
+
Population = gr.Number(label = "Enter Population")
|
21 |
+
AveOccup = gr.Number(label = "Enter AveOccup")
|
22 |
+
Latitude = gr.Number(label = "Enter Latitude")
|
23 |
+
Longitude = gr.Number(label = "Enter Longitude")
|
24 |
+
# We create the output
|
25 |
+
output = gr.Textbox()
|
26 |
+
|
27 |
+
|
28 |
+
app = gr.Interface(fn = make_prediction, inputs=[MedInc,HouseAge,AveRooms,AveBedrms,Population,AveOccup,Latitude,Longitude], outputs=output)
|
29 |
+
app.launch()
|
filename.pkl
ADDED
Binary file (719 Bytes). View file
|
|
scaler.pkl
ADDED
Binary file (1.22 kB). View file
|
|