Update app.py

app.py

@@ -163,13 +163,17 @@ def keywords(text,prompt_num,user_keywords):
     count = 0
     images = []
     # np_images = []
+    print("works1")
     while count != int(len(generated_prompts)):
 
-      for i in generated_prompts:
-        count += 1
-        print(i)
-        image = pipe(i, height=768, width=768, guidance_scale = 10).images[0]
-        images.append(image)
+        print("works2")
+
+        for i in generated_prompts:
+            print("works3")
+            count += 1
+            print(i)
+            image = pipe(i, height=768, width=768, guidance_scale = 10).images[0]
+            images.append(image)
     
     # 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])