Spaces:
Sleeping
Sleeping
| import sys | |
| sys.path.append('G:\Project\phishing-detection') | |
| from phishingdetection import FeatureExtraction | |
| import numpy as np | |
| from phishingdetection import gbc | |
| import streamlit as st | |
| st.title("Phishing Website Detection") | |
| # | |
| # User input for URL | |
| url = st.text_input("Enter the Url:", key="url_input") | |
| #can provide any URL. this URL was taken from PhishTank | |
| # Predict and display the result | |
| if st.button("Check"): | |
| if url: | |
| obj = FeatureExtraction(url) | |
| x = np.array(obj.getFeaturesList()).reshape(1, 30) | |
| y_pred = gbc.predict(x)[0] | |
| if y_pred == 1: | |
| st.write("We guess it is a safe website") | |
| else: | |
| st.write("Caution! Suspicious website detected") | |
| st.write(y_pred) | |
| else: | |
| st.write("Please enter a URL.") |