Update app.py

app.py

@@ -37,10 +37,22 @@ pickle.dump(modelodef, open("modelodef.pkl", "wb"))
 
 #create a function for gradio
 def RCS(JET, Suelo,SPT, WtoC, Presion, Velocidad):
-    x = np.array([JET, Suelo,SPT, WtoC, Presion, Velocidad])
-    prediction = modelodef.predict(x.reshape(1, -1))
+    modelodef = pickle.load(open("modelodef.pkl", "rb"))
+    prediction = modelodef.predict([[JET, Suelo,SPT, WtoC, Presion, Velocidad]])
     return prediction
 
-outputs = gr.outputs.Textbox()
-app = gr.Interface(fn=RCS, inputs=['number','number','number','number','number','number'], outputs=outputs,description="UCS jet grouting")
+
+app = gr.Interface(
+    RCS,
+    inputs=[
+        gr.Number(value=1, label="JET"),
+        gr.Number(value=1, label="Suelo"),
+        gr.Number(value=1, label="SPT"),
+        gr.Number(value=1, label="WtoC"),
+        gr.Number(value=1, label="Presion"),
+        gr.Number(value=1, label="velocidad"),
+    ],
+    outputs=[gr.Text(label="UCS")],
+)
+
 app.launch()