Update app.py

app.py

@@ -19,11 +19,11 @@ def infer(original_image):
     output_image = output_image.reshape(
         (np.shape(output_image)[0], np.shape(output_image)[1], 3)
     )
-    output_image = np.uint32(output_image)
+    output_image = np.uint8(output_image)
     return output_image
 
 iface = gr.Interface(
     fn=infer,
-    inputs=[gr.inputs.Image(label="image", type="pil", shape=(256, 256))],
+    inputs=[gr.inputs.Image(label="image", type="pil")],
     outputs="image",
     examples=examples).launch(debug=True)