Update inference_code.py

inference_code.py

@@ -17,7 +17,7 @@ def generate_images(prompt):
     num_inference_steps = 20
 
     images = pipeline(prompt, width=512, num_inference_steps=20, num_images_per_prompt=1).images
-    images = pipeline.numpy_to_pil(np.asarray(images.reshape((1,) + images.shape[-3:])))
+    # images = pipeline.numpy_to_pil(np.asarray(images.reshape((1,) + images.shape[-3:])))
 
     
     # num_samples = jax.device_count()