Update app.py

app.py

@@ -13,34 +13,31 @@ st.markdown(
     "Enter a legitimate URL and a potentially typosquatted URL to see the classifier's prediction."
 )
 
-# Load the classifier model from Hugging Face
-model_path = "./URLGuardian"  # Model repository on Hugging Face
-st.markdown(model_path
-)
+model_path = "./URLGuardian" 
 classifier = load_classifier(model_path)
 
 # URL inputs
 url = st.text_input("Enter the URL:", value="https://example.com")
 
 # Typosquatting detection on button click
-if st.button("Check Typosquatting"):
-    if legit_url and suspect_url:
-        
+if st.button("Check Safety of the url"):
+    if url:
+        # Run the classifier on the input URL
         result = classifier(url)[0]
-        label = result['label']
-        score = result['score']
-        
-        # Adjust the label names as per the model's documentation.
-        # This example assumes the label for a typosquatted URL might include "typo".
-        if "typo" in label.lower():
+        label = result["label"]
+        score = result["score"]
+
+        # Display result based on the label
+        # Adjust the label checking logic based on the model's documentation.
+        if "safe" in label.lower():
             st.success(
-                f"The model predicts that '{suspect_url}' is likely a typosquatted version of '{legit_url}' "
-                f"with a confidence of {score * 100:.2f}%."
+                f"The URL '{url}' is considered safe with a confidence of {score * 100:.2f}%."
             )
         else:
-            st.warning(
-                f"The model predicts that '{suspect_url}' is NOT likely a typosquatted version of '{legit_url}' "
-                f"with a confidence of {score * 100:.2f}%."
+            st.error(
+                f"The URL '{url}' is considered suspicious with a confidence of {score * 100:.2f}%."
             )
+        # Optionally, you can display the full result for debugging purposes:
+        st.write("Full classification output:", result)
     else:
-        st.error("Please enter both a legitimate URL and a potentially typosquatted URL.")
+        st.error("Please enter a URL.")