Update app.py

app.py

@@ -147,10 +147,11 @@ def checkbox_block():
   return checkbox
 
 def infer(text):
-  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"]):