Update app.py

app.py

@@ -4,14 +4,13 @@ from tensorflow.keras.preprocessing.image import load_img, img_to_array
 from tensorflow.keras.models import load_model
 from PIL import Image
 
-inputs = gr.inputs.Image()
+inputs = gr.inputs.Image(shape=(512, 512))
 o1 = gr.outputs.Image()
 o2 = gr.outputs.Image()
 gen_model = load_model('generator_model.h5')
 
 def colorify(pixels):
   
-  pixels = np.reshape(pixels, (1, 512, 512, 3))
   pixels = (pixels - 127.5) / 127.5
   pixels = np.expand_dims(pixels, 0)
   gen_image = gen_model.predict(pixels)