Update main.py

main.py

@@ -10,7 +10,7 @@ app.secret_key = "cricktech-abhi-2023"
 CORS(app)
 
 # Load the model
-model = joblib.load('cricket_world_cup_score_prediction_model.pkl')
+model = joblib.load('dataset/world/cricket_world_cup_score_prediction_model.pkl')
 
 
 @app.route('/', methods=['GET'])
@@ -25,7 +25,7 @@ def prediction():
         batting_team = data.get('batting_team')
         bowling_team = data.get('bowling_team')
         score = score_predict(batting_team, bowling_team, runs=data.get('total_runs'), wickets=data.get('total_wickets'), overs=data.get('overs'), runs_last_5_overs=data.get('runs_last_5_overs'), wickets_last_5_overs=data.get('wickets_last_5_overs'))
-        return score
+        return {'success': True, 'prediction': str(score)}
 
 
 def score_predict(batting_team, bowling_team, runs, wickets, overs, runs_last_5_overs, wickets_last_5_overs):