EstebanDC commited on
Commit
0d9c93e
1 Parent(s): 9f7459c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -38,8 +38,8 @@ pickle.dump(modelodef, open("modelodef.pkl", "wb"))
38
  #create a function for gradio
39
  def RCS(JET, Suelo,SPT, WtoC, Presion, Velocidad):
40
  modelodef = pickle.load(open("modelodef.pkl", "rb"))
41
- prediction = modelodef.predict([[JET, Suelo,SPT, WtoC, Presion, Velocidad]])
42
- prediction = np.round(prediction,2)
43
  return prediction
44
 
45
  title = "UCS"
 
38
  #create a function for gradio
39
  def RCS(JET, Suelo,SPT, WtoC, Presion, Velocidad):
40
  modelodef = pickle.load(open("modelodef.pkl", "rb"))
41
+ prediction0 = modelodef.predict([[JET, Suelo,SPT, WtoC, Presion, Velocidad]])
42
+ prediction = np.round(prediction0,2)
43
  return prediction
44
 
45
  title = "UCS"