Streamlit_code / app.py
Junaid-EEE11's picture
Update app.py
fc54e2d
# app.py
import streamlit as st
from models.fraud_detection_model import load_fraud_detection_model, predict_fraud
from utils.preprocessing import preprocess_data_for_streamlit
from utils import feature_engineering
import pandas as pd
import tensorflow
def load_parquet_file(parquet_file_path):
return pd.read_parquet(parquet_file_path)
model_path = 'models/fraud_detection_model.h5'
fraud_model = load_fraud_detection_model(model_path)
from datasets import load_dataset
dataset = load_dataset("iix/Parquet_FIles/Fraud_detection.parquet")
# Load data
#data_path = 'data/dataset.csv'
df, X_scaled = preprocess_data_for_streamlit(dataset)
# Streamlit App
st.title('Fraud Detection Web App')
# Sidebar with user input
selected_index = st.sidebar.selectbox('Select an index:', df.index)
selected_data = X_scaled[selected_index].reshape(1, -1)
# Display selected data
st.write('Selected Data:')
st.write(df.iloc[selected_index])
# Predict fraud
if st.button('Predict Fraud'):
prediction = predict_fraud(fraud_model, selected_data)
result = "Fraud" if prediction[0][0] == 1 else "Non-Fraud"
st.write(f'Prediction: {result}')