Update app.py

app.py

@@ -173,7 +173,8 @@ def keywords(text,prompt_num,user_keywords):
             count += 1
             print(i)
             print("works4")
-            image = pipe(i, height=768, width=768, guidance_scale = 10).images[0]
+            with torch.autocast("cuda"):
+                image = pipe(i, height=768, width=768, guidance_scale = 10).images[0]
             print("works5")
             images.append(image)
             print("works6")