I have to reshape the array

model.py

@@ -30,9 +30,9 @@ class SmokerModel:
 
         return new_data_scaled
 
-    def predict(self, X):
+    def predict(self, X: np.ndarray):
         """
-        Make predictions using the loaded model.
+        Make a prediction on one sample using the loaded model.
 
         INPUT
         -----
@@ -43,6 +43,14 @@ class SmokerModel:
         predicted label
         """
 
+        # Check if the array is 1-dimensional aka one sample
+        if len(X.shape) != 1:
+            raise ValueError("Input array must be one-dimensional (one sample), but got a shape of {}".format(X.shape))
+            return
+            
+        # Reshape the array
+        X = X.reshape(1, -1)  
+
         # scale the data
         X_scaled = self.scale(X)