Update app.py

app.py

@@ -9,12 +9,14 @@ st.write("Enter two domains to check if one is a typosquatted variant of the oth
 
 domain = st.text_input("Enter the legitimate domain name:")
 sim_domain = st.text_input("Enter the potentially typosquatted domain name:")
+threshold = st.slider("Set detection threshold", 0.0, 1.0, 0.5)
+
 
 if st.button("Check Typosquatting"):
     inputs = [(domain, sim_domain)]
     prediction = model.predict(inputs)[0]
     print(prediction)
-    if prediction > 0.5:
-        st.success(f"The model predicts that '{sim_domain}' is likely a typosquatted version of '{domain}' with {prediction}.")
+    if prediction > threshold:
+        st.success(f"The model predicts that '{sim_domain}' is likely a typosquatted version of '{domain}' with a score of {prediction}.")
     else:
-        st.warning(f"The model predicts that '{sim_domain}' is NOT likely a typosquatted version of '{domain}' with {prediction}.")
+        st.warning(f"The model predicts that '{sim_domain}' is NOT likely a typosquatted version of '{domain}' with a score of {prediction}.")