Update app.py

app.py

@@ -42,7 +42,7 @@ def process_image(image):
     img /= 255.0
     img = np.expand_dims(img, axis=0)
 
-    layer_model = Model(inputs=autoencoder.input, outputs=autoencoder.layers[layer].output)
+    layer_model = Model(inputs=autoencoder.input, outputs=autoencoder.layers[4].output)
 
     encoded_array = layer_model.predict(img)