Update app.py

app.py

@@ -276,7 +276,7 @@ def generate_30(
     print('catted pooled shape 2: ', pooled_prompt_embeds2.shape)
     pooled_prompt_embeds2 = torch.mean(pooled_prompt_embeds2,dim=0,keepdim=True)
     print('pooled meaned shape 2: ', pooled_prompt_embeds2.shape)
-    pooled_prompt_embeds = torch.cat([pooled_prompt_embeds, pooled_prompt_embeds2],dim=1)
+    pooled_prompt_embeds = pooled_prompt_embeds2 #torch.cat([pooled_prompt_embeds, pooled_prompt_embeds2],dim=1)
     print('catted combined meaned pooled shape: ', pooled_prompt_embeds.shape)
   
     prompt_embeds = torch.cat([prompt_embeds, prompt_embeds2],dim=-1)
@@ -411,7 +411,7 @@ def generate_60(
     print('catted pooled shape 2: ', pooled_prompt_embeds2.shape)
     pooled_prompt_embeds2 = torch.mean(pooled_prompt_embeds2,dim=0,keepdim=True)
     print('pooled meaned shape 2: ', pooled_prompt_embeds2.shape)
-    pooled_prompt_embeds = torch.cat([pooled_prompt_embeds, pooled_prompt_embeds2],dim=1)
+    pooled_prompt_embeds = pooled_prompt_embeds2 #torch.cat([pooled_prompt_embeds, pooled_prompt_embeds2],dim=1)
     print('catted combined meaned pooled shape: ', pooled_prompt_embeds.shape)
   
     prompt_embeds = torch.cat([prompt_embeds, prompt_embeds2],dim=-1)
@@ -546,7 +546,7 @@ def generate_90(
     print('catted pooled shape 2: ', pooled_prompt_embeds2.shape)
     pooled_prompt_embeds2 = torch.mean(pooled_prompt_embeds2,dim=0,keepdim=True)
     print('pooled meaned shape 2: ', pooled_prompt_embeds2.shape)
-    pooled_prompt_embeds = torch.cat([pooled_prompt_embeds, pooled_prompt_embeds2],dim=1)
+    pooled_prompt_embeds = pooled_prompt_embeds2 #torch.cat([pooled_prompt_embeds, pooled_prompt_embeds2],dim=1)
     print('catted combined meaned pooled shape: ', pooled_prompt_embeds.shape)
   
     prompt_embeds = torch.cat([prompt_embeds, prompt_embeds2],dim=-1)