Spaces:
Sleeping
Sleeping
import gradio as gr | |
from featureExtractor import featureExtraction | |
from pycaret.classification import load_model, predict_model | |
model = load_model('phishingdetection') | |
def predict(url): | |
data = featureExtraction(url) | |
result = predict_model(model, data = data) | |
return result['prediction_label'][0] | |
app = gr.Interface(fn=predict, | |
inputs="text", | |
outputs="text", | |
) | |
if __name__ == "__main__": | |
app.launch(debug = True) | |