Spaces:
Build error
Build error
| import gradio as gr | |
| import pickle | |
| # Load the model and the scaler | |
| with open("fraud_detection_model.pkl", "rb") as f: | |
| model = pickle.load(f) | |
| with open("scaler.pkl", "rb") as f: | |
| scaler = pickle.load(f) | |
| # Define a function for prediction | |
| def predict(input_data): | |
| input_data = [float(x) for x in input_data.split(",")] | |
| input_data = scaler.transform([input_data]) # Apply the scaling transformation | |
| prediction = model.predict(input_data) | |
| return "Fraud" if prediction[0] == 1 else "Not Fraud" | |
| # Updated Gradio Interface | |
| input_text = gr.Textbox(label="Input Features (comma-separated)", placeholder="Enter features like 1.2, 3.4, ...") | |
| output_text = gr.Textbox(label="Prediction") | |
| interface = gr.Interface( | |
| fn=predict, | |
| inputs=input_text, | |
| outputs=output_text, | |
| title="Fraud Detection System", | |
| description="Enter the features to predict whether it is fraud or not." | |
| ) | |
| # Run the application | |
| interface.launch() |