Upload app.py

app.py

@@ -49,24 +49,18 @@ app = FastAPI(title="Medicare Fraud Prediction API")
 
 @app.post("/predict")
 def predict_fraud(data: ClaimData):
+    # Convert input to DataFrame
     df = pd.DataFrame([data.dict()])
 
-    cols = ['OPAnnualReimbursementAmt',
-            'OPAnnualDeductibleAmt',
-            'DeductibleAmtPaid']
-
-    # build a fresh DataFrame with ONLY these columns & names
-    scaled = pd.DataFrame(
-        scaler.transform(df[cols].to_numpy()),
-        columns=cols
-    )
-
-    # replace the three columns with their scaled versions
-    for c in cols:
-        df[c] = scaled[c]
+    # Scale only the relevant columns
+    columns_to_scale = ['OPAnnualReimbursementAmt','OPAnnualDeductibleAmt','DeductibleAmtPaid']
+    df[columns_to_scale] = scaler.transform(df[columns_to_scale])
 
+    # Predict
     pred = model.predict(df)[0]
-    return {"prediction": "Fraud" if pred == 1 else "Not Fraud"}
+    result = "Fraud" if pred == 1 else "Not Fraud"
+    
+    return {"prediction": result}
 
 @app.get("/")
 def root():