vmonney's picture
Upload the application files
f80bc09
# 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()