Update main.py

main.py

@@ -3,11 +3,17 @@ from pydantic import BaseModel
 from fastapi import FastAPI
 import uvicorn
 import logging
+import pickle
+
+
+
+# Save using pickle
 
 logging.basicConfig(level = logging.INFO)
 # 1. Load the trained model
 model = joblib.load('frauddetection.pkl')
-
+with open('frauddetection.pkl', 'wb') as f:
+    pickle.dump(model, f, protocol=pickle.HIGHEST_PROTOCOL)
 # 2. Define the input data schema using Pydantic BaseModel
 class InputData(BaseModel):
     Year:int