Update app.py

app.py

@@ -1,5 +1,7 @@
 import numpy as np
 import gradio as gr
+from PIL import Image
+import keras
 from huggingface_hub import from_pretrained_keras
 
 
@@ -7,7 +9,9 @@ model = from_pretrained_keras("keras-io/lowlight-enhance-mirnet", compile=False)
 examples = ['examples/179.png', 'examples/493.png', 'examples/780.png']
 
 
-def infer(image):
+def infer(original_image):
+    image = 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)
     output_image = output[0] * 255.0
@@ -15,9 +19,9 @@ def infer(image):
     output_image = output_image.reshape(
         (np.shape(output_image)[0], np.shape(output_image)[1], 3)
     )
+    output_image = Image.fromarray(np.uint8(output_image))
     return output_image
 
-
 iface = gr.Interface(
     fn=infer,
     inputs=[gr.inputs.Image(label="image", type="pil", shape=(128, 128))],