Update main.py

main.py

@@ -8,7 +8,9 @@ os.environ.setdefault("TOKENIZERS_PARALLELISM", "false")
 
 URL_MODEL_ID = "CrabInHoney/urlbert-tiny-v4-malicious-url-classifier"
 LABEL_MAP = {0: "benign", 1: "defacement", 2: "malware", 3: "phishing"}
-URL_RE = re.compile(r"""(?xi)\b(?:https?://|www\.)[a-z0-9\-._~%]+(?:/[^\s<>"']*)?""")
+#URL_RE = re.compile(r"""(?xi)\b(?:https?://|www\.)[a-z0-9\-._~%]+(?:/[^\s<>"']*)?""")
+URL_RE = re.compile(r"""(?xi)\bhttps?://[^\s<>"']+""")
+
 
 _tok = None
 _mdl = None