ling360demo / app.py
webomurga's picture
Update app.py
394e645 verified
import streamlit as st
import pickle
model = pickle.load(open("phishing_test.pkl", "rb"))
st.title("URL Checker")
st.subheader("Just type a URL below to see if it's good!")
url_to_predict = st.text_input("", value="google.com", placeholder="Enter URL")
prediction = model.predict([url_to_predict])
st.write(f"The URL you've entered is {prediction[0]}!")