gradio/demo/fraud_detector/run.py
Ömer Faruk Özdemir cc0cff893f Format The Codebase
- black formatting
- isort formatting
2022-01-21 16:44:12 +03:00

39 lines
994 B
Python

import random
import pandas as pd
import gradio as gr
def fraud_detector(card_activity, categories, sensitivity):
activity_range = random.randint(0, 100)
drop_columns = [
column for column in ["retail", "food", "other"] if column not in categories
]
if len(drop_columns):
card_activity.drop(columns=drop_columns, inplace=True)
return (
card_activity,
card_activity,
{"fraud": activity_range / 100.0, "not fraud": 1 - activity_range / 100.0},
)
iface = gr.Interface(
fraud_detector,
[
gr.inputs.Timeseries(x="time", y=["retail", "food", "other"]),
gr.inputs.CheckboxGroup(
["retail", "food", "other"], default=["retail", "food", "other"]
),
gr.inputs.Slider(1, 3),
],
[
"dataframe",
gr.outputs.Timeseries(x="time", y=["retail", "food", "other"]),
gr.outputs.Label(label="Fraud Level"),
],
)
if __name__ == "__main__":
iface.launch()