DanielViniciusAlves commited on
Commit
91836b2
·
1 Parent(s): 4112b77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -8
app.py CHANGED
@@ -1,13 +1,19 @@
1
  import gradio as gr
2
  import skimage
3
  import pickle
 
4
 
5
 
6
  with open('model.pkl', 'rb') as f:
7
  model = pickle.load(f)
8
 
9
- def predict(input):
10
- predictions = model.predict(input)
 
 
 
 
 
11
  return {predictions}
12
 
13
  gr.Interface(
@@ -15,13 +21,13 @@ gr.Interface(
15
  title="Wine Quality Model",
16
  allow_flagging="never",
17
  inputs=[
18
- gr.inputs.Number(default=7.3, label="fixed acidity"),
19
- gr.inputs.Number(default=0.27, label="volatile acidity"),
20
- gr.inputs.Number(default=0.3, label="citric acid"),
21
- gr.inputs.Number(default=1.5, label="residual sugar"),
22
  gr.inputs.Number(default=0.08, label="chlorides"),
23
- gr.inputs.Number(default=20, label="free sulfur dioxide"),
24
- gr.inputs.Number(default=50, label="total sulfur dioxide"),
25
  gr.inputs.Number(default=0.998, label="density"),
26
  gr.inputs.Number(default=3.5, label="pH"),
27
  gr.inputs.Number(default=0.6, label="sulphates"),
 
1
  import gradio as gr
2
  import skimage
3
  import pickle
4
+ import pandas as pd
5
 
6
 
7
  with open('model.pkl', 'rb') as f:
8
  model = pickle.load(f)
9
 
10
+ def predict(fixedAcidity, volatileAcidity, citricAcid, residualSugar, chlorides, freeSulfurDioxide, totalSulfurDioxide, density, pH, sulphates, alcohol, LogTotalSulfurDioxide):
11
+
12
+ data = [[float(fixedAcidity), float(volatileAcidity), float(citricAcid), float(residualSugar), float(chlorides), float(freeSulfurDioxide), float(totalSulfurDioxide), float(density), float(pH), float(sulphates), float(alcohol), float(LogTotalSulfurDioxide)]]
13
+
14
+ row_df=pd.DataFrame(data,columns=['fixed acidity','volatile acidity', 'citric acid', 'residual sugar', 'chlorides', 'free sulfur dioxide', 'total sulfur dioxide', 'density', 'pH', 'sulphates', 'alcohol', 'LogTotalSulfurDioxide'])
15
+
16
+ predictions = model.predict(row_df)
17
  return {predictions}
18
 
19
  gr.Interface(
 
21
  title="Wine Quality Model",
22
  allow_flagging="never",
23
  inputs=[
24
+ gr.inputs.Number(default=7.3, label="fixedAcidity"),
25
+ gr.inputs.Number(default=0.27, label="volatileAcidity"),
26
+ gr.inputs.Number(default=0.3, label="citricAcid"),
27
+ gr.inputs.Number(default=1.5, label="residualSugar"),
28
  gr.inputs.Number(default=0.08, label="chlorides"),
29
+ gr.inputs.Number(default=20, label="freeSulfurDioxide"),
30
+ gr.inputs.Number(default=50, label="totalSulfurDioxide"),
31
  gr.inputs.Number(default=0.998, label="density"),
32
  gr.inputs.Number(default=3.5, label="pH"),
33
  gr.inputs.Number(default=0.6, label="sulphates"),