Spaces:
Runtime error
Runtime error
import gradio as gr | |
import pandas as pd | |
import joblib | |
model = joblib.load('decision_tree.pkl') | |
def onehot(df, column): | |
df = df.copy() | |
dummies = pd.get_dummies(df[column], prefix='type') | |
df = pd.concat([df,dummies], axis=1) | |
df = df.drop(column, axis=1) | |
return df | |
def dataframe(file_obj): | |
df = pd.read_csv(file_obj.name) | |
df = onehot(df, column='type') | |
df = df.drop(['nameOrig','nameDest'], axis=1) | |
print(df.shape) | |
y_pred = model.predict(df) | |
pred_df = pd.DataFrame(y_pred) | |
print(type(pred_df)) | |
print(pred_df.shape) | |
# clr = classification_report(y_test, y_pred, target_names=['Not Fraud','Fraud']) | |
# return 'Classification Report:\n'+ clr | |
return pred_df | |
file = gr.inputs.File(file_count="single", type="file", label="CSV File for Predictions", optional=False) | |
y_pred_df = gr.outputs.Dataframe(max_rows=20, max_cols=None, overflow_row_behaviour="paginate", type="pandas", label="Predictions of records in the file") | |
interface_csv = gr.Interface( | |
fn=dataframe, | |
inputs=file, | |
outputs=y_pred_df, | |
title="Fraud Detection - EXPERT SYSTEM", | |
theme="dark-peach" | |
) | |
interface_csv.launch(inline=False) |