Update app.py

app.py

@@ -40,9 +40,13 @@ def process_image(image):
     img = img.astype('float32')  
     img /= 255.0
     img = np.expand_dims(img, axis=0)
-    encoded_features = autoencoder.predict(img)
 
-    return encoded_features
+    layer_model = Model(inputs=model.input, outputs=model.layers[layer].output)
+
+    encoded_array = layer_model.predict(img) 
+
+    pooled_array = encoded_array.max(axis=-1)
+    return pooled_array  # Shape: (1, n_features)
     
 def inference(image):