Cropinky commited on
Commit
2562027
1 Parent(s): d16841b

bug fix for generating only 1 image

Browse files
Files changed (1) hide show
  1. image_generator.py +1 -1
image_generator.py CHANGED
@@ -144,7 +144,7 @@ def generate_images(
144
  #imgs_complete = np.vstack(np.hsplit(imgs_row, 4))
145
  #cv2.imshow("lalaxd", imgs_complete)
146
  #cv2.waitKey()
147
- return PIL.Image.fromarray(img, 'RGB')
148
 
149
 
150
  #----------------------------------------------------------------------------
144
  #imgs_complete = np.vstack(np.hsplit(imgs_row, 4))
145
  #cv2.imshow("lalaxd", imgs_complete)
146
  #cv2.waitKey()
147
+ return PIL.Image.fromarray(img[0].cpu().numpy(), 'RGB')
148
 
149
 
150
  #----------------------------------------------------------------------------