Spaces:
Runtime error
Runtime error
RamAnanth1
commited on
Commit
•
e2e2967
1
Parent(s):
9022b97
Update model.py
Browse files
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]
|