Update app.py

app.py

@@ -25,12 +25,12 @@ nltk.download('wordnet')
 STOPWORDS = set(stopwords.words('english'))
 lemmatizer = WordNetLemmatizer()
 
-def normalize_length(url, target_length=50):
-    if len(url) < target_length:
-        url = url + " " * (target_length - len(url))
+def normalize_length(text, target_length=50):
+    if len(text) < target_length:
+        text = text + " " * (target_length - len(text))
     else:
-        url = url[:target_length]
-    return url
+        text = text[:target_length]
+    return text
 
 def preprocess_url(url):
     url = url.lower()
@@ -86,8 +86,7 @@ def get_prediction(input_text, input_type):
 
 def phishing_detection(input_text, input_type):
     prediction = get_prediction(input_text, input_type)
-    threshold = 0.5  # Adjusted threshold
-    if prediction > threshold:
+    if prediction > 0.5:
         return f"Warning: This site is likely a phishing site! ({prediction:.2f})"
     else:
         return f"Safe: This site is not likely a phishing site. ({prediction:.2f})"
@@ -104,4 +103,4 @@ iface = gr.Interface(
     theme="default"
 )
 
-iface.launch()
+iface.launch()