albin commited on
Commit
e9ff521
·
1 Parent(s): 8ec8995
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -126,7 +126,7 @@ async def predict(request: Request, requess: Req = Depends(form_req)):
126
  print("Data sent to predict:", data)
127
  print("Data types:", dataFrame_features[selected_columns].dtypes)
128
 
129
- prediction = model.predict([data])
130
  output = prediction[0]
131
 
132
  output_text = "Legitimate" if output == 1 else "Phishing"
 
126
  print("Data sent to predict:", data)
127
  print("Data types:", dataFrame_features[selected_columns].dtypes)
128
 
129
+ prediction = model.predict(data)
130
  output = prediction[0]
131
 
132
  output_text = "Legitimate" if output == 1 else "Phishing"