RamAnanth1 commited on
Commit
e2e2967
1 Parent(s): 9022b97

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -366,4 +366,4 @@ class Model:
366
  x_samples_ddim = 255.*x_samples_ddim
367
  x_samples_ddim = x_samples_ddim.astype(np.uint8)
368
 
369
- return [im_pose, x_samples_ddim]
 
366
  x_samples_ddim = 255.*x_samples_ddim
367
  x_samples_ddim = x_samples_ddim.astype(np.uint8)
368
 
369
+ return [im_pose[:,:,::-1].astype(np.uint8), x_samples_ddim]