Update app.py

app.py

@@ -2,13 +2,14 @@ import numpy as np
 import gradio as gr
 from huggingface_hub import from_pretrained_keras
 import keras
+import tensorflow as tf
 
 model = from_pretrained_keras("keras-io/lowlight-enhance-mirnet", compile=False)
 examples = ['examples/1.png', 'examples/55.png', 'examples/665.png', 'examples/778.png']
 
 
 def infer(original_image):
-    image = keras.preprocessing.image.img_to_array(original_image)
+    image = tf.keras.preprocessing.image.img_to_array(original_image)
     image = image.astype("float32") / 255.0
     image = np.expand_dims(image, axis=0)
     output = model.predict(image)