bug fix

app.py

@@ -150,12 +150,13 @@ def predict_and_analyze(model_name, num_channels, dim, image):
     
     image = image.reshape((num_channels, W, W))
     print(image)
+    print(type(image))
     
     # W = int(np.sqrt(image.shape[1]))
     
     # image = image.reshape((num_channels, W, W))
     
-    if len(image.shape != 4):
+    if len(image.shape) != 4:
         image = image[np.newaxis, :, :, :]
         
     input_image = np.sum(image[0, :, :, :], axis=0)