reshape to 32

app.py

@@ -150,14 +150,14 @@ def encrypt(user_id, input_image):
 
     import numpy as np
     # Resize image if necessary
-    if input_image.shape != (224, 224, 3):
+    if input_image.shape != (32, 32, 3):
         input_image_pil = Image.fromarray(input_image)
-        input_image_pil = input_image_pil.resize((224, 224))
+        input_image_pil = input_image_pil.resize((32, 32))
         input_image = np.array(input_image_pil)
 
     # Convert to grayscale and reshape to (1, 1, 224, 224)
     input_image = np.mean(input_image, axis=2).astype(np.float32)
-    input_image = input_image.reshape(1, 1, 224, 224)
+    input_image = input_image.reshape(1, 1, 32, 32)
 
     # Scale values to 12-bit range (-2048 to 2047)
     input_image = (input_image / 255.0 * 4095 - 2048).astype(np.int16)