Update app.py

app.py

@@ -146,12 +146,12 @@ def checkbox_block():
   return checkbox
 
 def infer(text):
-  #with autocast("cuda"):
-  images_list = pipe(
-            [text]*2,
-            num_inference_steps=50,
-            guidance_scale=7.5
-  )
+  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)