Update app.py

app.py

@@ -23,7 +23,8 @@ def infer(original_image):
     output_image = np.uint32(output_image)'''
     
     output_image = tf.cast((output[0, :, :, :] * 255), dtype=np.uint8)
-    output_image = Image.fromarray(output_image.numpy())
+    #output_image = Image.fromarray(output_image.numpy())
+    output_image = output_image.numpy()
     return output_image
     
 
@@ -32,7 +33,7 @@ iface = gr.Interface(
     title="Low Light Image Enhancement",
     description = "Keras Implementation of MIRNet model for light up the dark image 🌆🎆",
     inputs=[gr.inputs.Image(label="image", type="pil")],
-    outputs="image",
+    outputs=[gr.outputs.Image(label="image", type="numpy")],
     #examples=examples,
     article = "Author: <a href=\"https://huggingface.co/vumichien\">Vu Minh Chien</a>. Based on the keras example from <a href=\"https://keras.io/examples/vision/mirnet/\">Soumik Rakshit</a>",
     ).launch(debug=True)