Update app.py

app.py

@@ -214,10 +214,12 @@ def generate_30(
     prompt_embedsb = pipe.text_encoder(text_input_ids2.to(device), output_hidden_states=True)
     prompt_embedsb = prompt_embedsb.hidden_states[-2]
     prompt_embeds = torch.cat([prompt_embedsa,prompt_embedsb]).mean(dim=-1)
-
+    pooled_prompt_embeds = prompt_embeds[0]
+    
     options = {
         #"prompt": prompt,
         "prompt_embeds": prompt_embeds,
+        "pooled_prompt_embeds": pooled_prompt_embeds,
         "negative_prompt": negative_prompt,
         "negative_prompt_2": neg_prompt_2,
         "width": width,
@@ -281,10 +283,12 @@ def generate_60(
     prompt_embedsb = pipe.text_encoder(text_input_ids2.to(device), output_hidden_states=True)
     prompt_embedsb = prompt_embedsb.hidden_states[-2]
     prompt_embeds = torch.cat([prompt_embedsa,prompt_embedsb]).mean(dim=-1)
-
+    pooled_prompt_embeds = prompt_embeds[0]
+    
     options = {
         #"prompt": prompt,
         "prompt_embeds": prompt_embeds,
+        "pooled_prompt_embeds": pooled_prompt_embeds,
         "negative_prompt": negative_prompt,
         "negative_prompt_2": neg_prompt_2,
         "width": width,
@@ -348,10 +352,12 @@ def generate_90(
     prompt_embedsb = pipe.text_encoder(text_input_ids2.to(device), output_hidden_states=True)
     prompt_embedsb = prompt_embedsb.hidden_states[-2]
     prompt_embeds = torch.cat([prompt_embedsa,prompt_embedsb]).mean(dim=-1)
-
+    pooled_prompt_embeds = prompt_embeds[0]
+    
     options = {
         #"prompt": prompt,
         "prompt_embeds": prompt_embeds,
+        "pooled_prompt_embeds": pooled_prompt_embeds,
         "negative_prompt": negative_prompt,
         "negative_prompt_2": neg_prompt_2,
         "width": width,