Update app.py

app.py

@@ -167,16 +167,16 @@ def infer(text):
 
 # idetnical to `infer` function without gradio state updates for share btn
 def infer_examples(text):
-  with autocast("cuda"):
-    images_list = pipe(
+  #with autocast("cuda"):
+  images_list = pipe(
               [text]*2,
               num_inference_steps=50,
               guidance_scale=7.5
   )
-  output_images = []
-  for i, image in enumerate(images_list["sample"]):
-    output_images.append(image)
-  return output_images
+  #output_images = []
+  #for i, image in enumerate(images_list["sample"]):
+  #  output_images.append(image)
+  return images_list.images
   
 css = '''
 .gradio-container {font-family: 'IBM Plex Sans', sans-serif}