Update app.py

app.py

@@ -9,10 +9,9 @@ o1 = gr.outputs.Image()
 o2 = gr.outputs.Image()
 gen_model = load_model('generator_model.h5')
 
-def colorify(inp):
+def colorify(pixels):
   
-  pixels = load_img(inp, target_size=(512, 512))
-  pixels = img_to_array(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)