Update app.py

app.py

@@ -52,8 +52,10 @@ def encode(prompt: str):
     embeds_2048 = proj(embeds_896)      # [1, L, 2048]
 
     # pooled -> moyenne
-    pooled = embeds_2048.mean(dim=1)     # [1, 2048]
-
+         # [1, 2048]
+    proj_pooled = nn.Linear(2048, 768)
+    pooled = proj_pooled(embeds_2048.mean(dim=1))
+    
     # Sauvegarde
     torch.save(embeds_2048, "embeds.pt")
     torch.save(pooled, "pooled.pt")