Spaces:
Build error
Build error
| import gradio as gr | |
| import pandas as pd | |
| import joblib | |
| # Load trained model | |
| model = joblib.load("fraud_model.pkl") | |
| def predict_fraud(file): | |
| df = pd.read_csv(file.name) | |
| predictions = model.predict(df.drop("Class", axis=1)) # Assuming "Class" is present | |
| df["Prediction"] = predictions | |
| return df[df["Prediction"] == 1] # Return only fraud | |
| interface = gr.Interface( | |
| fn=predict_fraud, | |
| inputs=gr.File(label="Upload CSV File"), | |
| outputs=gr.Dataframe(label="Fraudulent Transactions"), | |
| title="Fraud Detection System", | |
| description="Upload a bank statement (CSV) to detect fraudulent transactions." | |
| ) | |
| interface.launch() |