Update main.py

main.py

@@ -50,7 +50,7 @@ async def predict(data: InputData):
     # Convert the prediction to a string (or any other format you prefer)
     result = "Fraud" if prediction[0] == 1 else "Not a Fraud"
     # result = prediction 
-    print(prediction)
+    print(prediction[1])
 
     return {"prediction": result}
 # 4. Run the API with uvicorn