change procedure of prediction

model.py

@@ -48,13 +48,18 @@ class SmokerModel:
         X_scaled = self.scale(X)
 
         # Check if the array is 1-dimensional aka one sample
-        if len(X_scaled.shape) != 1:
-            raise ValueError("Input array must be one-dimensional (one sample), but got a shape of {}".format(X.shape))
+        # if len(X_scaled.shape) != 1:
+        #     raise ValueError("Input array must be one-dimensional (one sample), but got a shape of {}".format(X.shape))
+        #     return
+
+        #check array only has one sample
+        if X.shape[0] != 1:
+            raise ValueError("Input array must contain only one sample, but {} samples were found".format(X.shape[0]))
             return
             
         # Reshape the array
         # X = X.reshape(1, -1)  
-        X_scaled = X_scaled.reshape(1, -1) 
+        # X_scaled = X_scaled.reshape(1, -1) 
 
         # # scale the data
         # X_scaled = self.scale(X)