Update app.py

app.py

@@ -154,16 +154,16 @@ def checkbox_block():
   return checkbox
 
 def infer(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, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+  #output_images = []
+  #for i, image in enumerate(images_list.images):
+  #  output_images.append(image)
+  return images_list.images, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
 
 # idetnical to `infer` function without gradio state updates for share btn
 def infer_examples(text):