Update app.py

app.py

@@ -20,10 +20,11 @@ def predict(features):
     preds = model.predict(X)
     return float(preds[0])
 
-def main(PU_DO,trip_distance):
+def main(PULocationID,DOLocationID,trip_distance):
     """request input, preprocess it and make prediction"""
     input_data = {
-    "PU_DO": PU_DO,
+    "PULocationID": PULocationID,
+    "DOLocationID": DOLocationID,
     "trip_distance": trip_distance
     }
     features = prepare_features(input_data)
@@ -39,6 +40,7 @@ def main(PU_DO,trip_distance):
 #input
 input1 = gr.inputs.Number()
 input2 = gr.inputs.Number()
+input3 = gr.inputs.Number()
 
 #output object
 output = gr.outputs.Textbox() 
@@ -46,7 +48,7 @@ output = gr.outputs.Textbox()
 intf = gr.Interface(title = "New York taxi duration prediction",
                     description = "The objective of this project is to predict the duration of a taxi trip in the city of New York.",
                     fn=main, 
-                    inputs=[input1,input2], 
+                    inputs=[input1,input2,input3], 
                     outputs=[output], 
                     live=True,
                     enable_queue=True