bug fix for generating only 1 image

image_generator.py

@@ -144,7 +144,7 @@ def generate_images(
     #imgs_complete = np.vstack(np.hsplit(imgs_row, 4))
     #cv2.imshow("lalaxd", imgs_complete)
     #cv2.waitKey()
-    return PIL.Image.fromarray(img, 'RGB')
+    return PIL.Image.fromarray(img[0].cpu().numpy(), 'RGB')
 
 
 #----------------------------------------------------------------------------