clean code and modify threshold

main.py

@@ -97,37 +97,8 @@ async def predict(request: Request, requess: Req = Depends(form_req)):
 
     data = dataFrame_features[selected_columns].values.reshape(1, -1)
 
-    # data = []
-    # data.append(str(features['URL']))
-    # data.extend(int(features['URLLength']))
-    # data.extend(str(features['Domain']))
-    # data.extend(int(features['DomainLength']))
-    # data.extend(str(features['TLD']))
-    # data.extend(float(features['CharContinuationRate']))
-    # data.extend(int(features['TLDLength']))
-    # data.extend(int(features['NoOfSubDomain']))
-    # data.extend(float(features['DegitRatioInURL']))
-    # data.extend(float(features['SpacialCharRatioInURL']))
-    # data.extend(int(features['IsHTTPS']))
-    # data.append(features['URL'])
-    # data.append(features['URLLength'])
-    # data.append(features['Domain'])
-    # data.append(features['DomainLength'])
-    # data.append(features['TLD'])
-    # data.append(features['CharContinuationRate'])
-    # data.append(features['TLDLength'])
-    # data.append(features['NoOfSubDomain'])
-    # data.append(features['DegitRatioInURL'])
-    # data.append(features['SpacialCharRatioInURL'])
-    # data.append(features['IsHTTPS'])
-
-    # Check number of dimensions before prediction
-    print("Nb dimensions before prediction:", data.shape)
-    print("Data sent to predict:", data)
-    print("Data types:", dataFrame_features[selected_columns].dtypes)
-
     prediction_proba = model.predict_proba(data)[:, 1]
-    threshold = 0.05
+    threshold = 0.2
     output = 1 if prediction_proba >= threshold else 0
     
     output_text = "Legitimate" if output == 1 else "Phishing"