suhas1324's picture
Update app.py
850dc64 verified
Raw
History Blame Contribute Delete
949 Bytes
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()