reshape to have 2 dimensions for the prediction

main.py

@@ -95,7 +95,7 @@ async def predict(request: Request, requess: Req = Depends(form_req)):
         encoder = label_encoders[column]
         dataFrame_features[column] = dataFrame_features[column].apply(lambda x: safe_transform(encoder, x))
 
-    data = dataFrame_features[selected_columns].values
+    data = dataFrame_features[selected_columns].values.reshape(1, -1)
 
     # data = []
     # data.append(str(features['URL']))
@@ -121,6 +121,9 @@ async def predict(request: Request, requess: Req = Depends(form_req)):
     # data.append(features['SpacialCharRatioInURL'])
     # data.append(features['IsHTTPS'])
 
+    # Check number of dimensions before prediction
+    print("Nb dimensions before prediction:", data.shape)
+
     prediction = model.predict([data])
     output = prediction[0]