Update driverpospredictionfinal.py

driverpospredictionfinal.py

@@ -7,10 +7,9 @@ Original file is located at
     https://colab.research.google.com/drive/1DGTfO4HEZDof1phuficJD_J8v38JnF3C
 """
 
-
-
-
-
+from flask import Flask,jsonify
+import json
+app = Flask(__name__)
 
 
 from sklearn import tree, linear_model
@@ -85,7 +84,21 @@ def hyperamter_tuning_paramter_grid():
     }#'criterion': ['gini', 'entropy']
   return parameter_grid
 
-
+@app.route('/predict',methods=['POST']) # api endpoint -any url(requests) with /predict will route to the method 
+def prediction(req):
+  data = json.loads(req)
+  input_data = {}
+  input_data['LapNumber'] = float(data['lapNumber'])
+  input_data['LapTimes'] = float(data['LapTimes'])
+  input_data['PitStopTimes'] = float(data['PitStopTimes'])
+  input_data['PrevLap'] = float(data['PrevLap'])
+  input_data['AvgSpeed'] = float(data['AvgSpeed'])
+  input_data['AirTemp_Cel'] = float(data['AirTemp_Cel'])
+  input_data['TrackTemp_Cel'] = float(data['TrackTemp_Cel'])
+  input_data['Humidity'] = float(data['Humidity'])
+  input_data['WindSpeed_km'] = float(data['WindSpeed_km'])
+  print(input_data)
+  return jsonify({'message':'Hello world'})
 
 
 def ask_user(model, X_test=None):