Update app.py

app.py

@@ -16,14 +16,14 @@ def defang_url(url: str) -> str:
     This function replaces the protocol and dots.
     For example:
         https://example.com  -->  hxxps://example[.]com
-    # """
-    # # Replace the protocol
-    # if url.startswith("https://"):
-    #     url = url.replace("https://", "hxxps://")
-    # elif url.startswith("http://"):
-    #     url = url.replace("http://", "hxxp://")
+    """
+    # Replace the protocol
+    if url.startswith("https://"):
+        url = url.replace("https://", "hxxps://")
+    elif url.startswith("http://"):
+        url = url.replace("http://", "hxxp://")
     
-    # # Replace periods in the rest of the URL
+    # Replace periods in the rest of the URL
     return url.replace(".", "[.]")
 
 st.title("URL Typosquatting Detection with URLGuardian")
@@ -35,7 +35,7 @@ st.markdown(
 model_path = "./URLGuardian" 
 classifier = load_classifier(model_path)
 
-url = st.text_input("Enter the URL:", value="https://example.com")
+url = st.text_input("Enter the URL:", value="example.com")
 
 if st.button("Check Safety of the url"):
     if url: