Image reconstruction using VAEs