Update app.py

app.py

@@ -7,8 +7,8 @@ import matplotlib.pyplot as plt
 
 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()
-o2 = gr.outputs.Image()
+o1 = gr.outputs.Image(type='numpy')
+o2 = gr.outputs.Image(type='numpy')
 gen_model = load_model('256_model_250ep.h5')
 
 def colorify(pixels):