Update app.py

app.py

@@ -176,11 +176,11 @@ def keywords(text,prompt_num,user_keywords):
             print(i)
             print("works4")
             torch.cuda.empty_cache()
-            with torch.autocast("cuda"):
-                image = pipe(i, height=768, width=768, guidance_scale = 10).images[0]
-                print("works5")
-                images.append(image)
-                print("works6")
+            # with torch.autocast("cuda"):
+            image = pipe(i, height=768, width=768, guidance_scale = 10).images[0]
+            print("works5")
+            images.append(image)
+            print("works6")
     
     # min_shape = sorted( [(np.sum(i.size), i.size ) for i in images])[0][1]
     # imgs_comb = np.hstack([i.resize(min_shape) for i in images])