Final Updates

main.py

@@ -1,10 +1,12 @@
 from fastapi import FastAPI,HTTPException
+from fastapi.middleware.cors import CORSMiddleware
 from typing import Literal,List
 import uvicorn
 from pydantic import BaseModel
 import pandas as pd
 import os
 import pickle
+from transformers import log_transform
 
 # setup
 SRC = os.path.abspath('.')
@@ -25,6 +27,14 @@ app = FastAPI(
     version= '1.0.0'
 )
 
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  
+    allow_credentials=True,
+    allow_methods=["*"],  
+    allow_headers=["*"],  
+)
+
 class IncomePredictionInput(BaseModel):
     age:                   int
     gender:                str
@@ -49,7 +59,7 @@ class IncomePredictionInput(BaseModel):
     losses:                int
     stocks_status:         int
     citizenship:           str
-    importance_of_record:  float
+
 
    
 class IncomePredictionOutput(BaseModel):
@@ -89,6 +99,5 @@ def income_classification(income: IncomePredictionInput):
         raise HTTPException(status_code=500, detail=f"Error during classification: {error_detail}")
 
 
-if __name__ == '__main__':
-    uvicorn.run('main:app', reload=True)
+
     

pipeline.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:dd28c23cc70beba35906a28c1b6937630ffb3bb4a7c5e8cb8276f66a33eb60e4
-size 4811
+oid sha256:5798dfdbe5793f5277903f528f13beb821822483d09768f514183eab7a6c7335
+size 5296

rfc_model.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cd6ad029aa941d54353a585b399b77c15380a06cfa80a2c81faf9697effe0aac
-size 267723561
+oid sha256:28b7b8c02725a3b1914fc171cdf0a03ac31360a2bc55515832273809804203ef
+size 353869321

transformers.py

@@ -0,0 +1,4 @@
+import numpy as np
+
+def log_transform(x):
+    return np.log(x + 1)