AnomalyGradio / app.py
nicknochnack's picture
Update app.py
b869164
raw
history blame contribute delete
651 Bytes
# Import dependencies
import gradio as gr
import numpy as np
import joblib
# Predict function
def predict(region1, region2, region3, region4, region5, region6):
model = joblib.load('./localoutlierfactor.joblib')
yhat = model.predict(np.array([[region1, region2, region3, region4, region5, region6]]))
return 'Weird Traffic' if yhat == -1 else 'Normal Traffic'
# Building the interface
app = gr.Interface(
title='Traffic Anomaly Detection 🚦',
fn=predict,
inputs=[gr.Slider(0,800),gr.Slider(0,800),gr.Slider(0,800),gr.Slider(0,800),gr.Slider(0,800),gr.Slider(0,800)],
outputs='text'
)
# Launch the app
app.launch()