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
 
-inputs = gr.inputs.Image(shape=(512, 512))
+inputs = gr.inputs.Image(shape=(256, 256))
 o1 = gr.outputs.Image()
 o2 = gr.outputs.Image()
-gen_model = load_model('generator_model.h5')
+gen_model = load_model('model256.h5')
 
 def colorify(pixels):