Update app.py

app.py

@@ -16,7 +16,7 @@ def autocontrast(tensor, cutoff=0):
     return adjusted_tensor
 
 def read_image(image):
-    tf.io.decode_image(image,channels=3)
+    image = tf.io.decode_image(image,channels=3)
     image = autocontrast(image)
     image.set_shape([None, None, 3])
     image = tf.cast(image, dtype=tf.float32) / 255
@@ -32,7 +32,7 @@ def enhance_image(input_image):
     # Process the input image using the loaded model
     image = read_image(input_image)
     image = np.expand_dims(image, axis=0)
-    output_image = model.predict(input_image)
+    output_image = model.predict(image)
     generated_image = np.squeeze(output_image, axis=0)
     generated_image = tf.keras.preprocessing.image.array_to_img(generated_image)
     # Return the output image