Update app.py

app.py

@@ -5,10 +5,10 @@ from tensorflow.keras.models import load_model
 from PIL import Image
 import matplotlib.pyplot as plt
 
-i1 = gr.inputs.Image(shape=(256, 256), type='pil')
+i1 = gr.inputs.Image(shape=(256, 256))
 #i2 = gr.inputs.Slider(minimum=2, maximum=4, step=0.1, default=None, label="Scale for intensity - the more value the less the intensity in the pixels")
-o1 = gr.outputs.Image(type="pil")
-o2 = gr.outputs.Image(type="pil")
+o1 = gr.outputs.Image()
+o2 = gr.outputs.Image()
 gen_model = load_model('256_model_250ep.h5')
 
 def colorify(pixels):