Update app.py

app.py

@@ -232,9 +232,9 @@ def generate_30(
     test_prompt_embeds = prompt_embeds.mean(dim=0,keepdim=True)
     print('averaged shape (keepdim): ', test_prompt_embeds.shape)
 
-    test_prompt_embeds_2 = torch.cat([prompt_embeds_a, prompt_embeds_b], dim=1).mean(dim=1)
+    test_prompt_embeds_2 = torch.cat([prompt_embeds_a, prompt_embeds_b], dim=0).mean(dim=1)
     print('averaged shape 2: ', test_prompt_embeds_2.shape)
-    test_prompt_embeds_3 = torch.cat([prompt_embeds_a, prompt_embeds_b]).mean(dim=1,keepdim=True)
+    test_prompt_embeds_3 = torch.cat([prompt_embeds_a, prompt_embeds_b]).mean(dim=0,keepdim=True)
     print('averaged shape 3(keepdim): ', test_prompt_embeds_3.shape)
     
     pooled_prompt_embeds = pooled_prompt_embeds.mean(dim=0)