Update app.py

app.py

@@ -5,16 +5,16 @@ from transformers import pipeline
 classifier = pipeline("text-classification", model='kithangw/phishing_link_detection')
 
 # Streamlit application title
-st.title("Please enter a suspicious email")
+st.title("Please enter a suspicious link")
 
-# Text input for user to enter the email to classify
-email = st.text_area("Enter the email to classify", "")
+# Text input for user to enter the link to classify
+email = st.text_area("Enter the link to classify", "")
 
 # Perform text classification when the user clicks the "Classify" button
 if st.button("Classify"):
-    if email:  # Check if email is not empty
-        # Perform text classification on the input email
-        results = classifier(email)
+    if link:  # Check if link is not empty
+        # Perform text classification on the input link
+        results = classifier(link)
         
         # The results variable contains a list with one item, which is a dictionary.
         # The dictionary has 'label' and 'score' as keys.
@@ -24,8 +24,8 @@ if st.button("Classify"):
 
         # Check the label and print out the corresponding message
         if label == "LABEL_1":  # Assuming LABEL_1 indicates phishing
-            st.write(f"The email you entered is {score}% likely to be a phishing email.")
+            st.write(f"The link you entered is {score}% likely to be a phishing email.")
         else:  # Assuming LABEL_0 indicates not phishing
-            st.write(f"The email you entered is {score}% likely to be not a phishing email.")
+            st.write(f"The link you entered is {score}% likely to be not a phishing email.")
     else:
-        st.error("Please enter an email to classify.")
+        st.error("Please enter an link to classify.")