Fix: return feature vectors for all images in the batch

src/CLIP.py

@@ -24,5 +24,5 @@ class CLIPImageEncoder:
             #image_features = self.model(**input).last_hidden_state[:,0].cpu()
             print(image_features)
             print("--------------------")
-            print(self.model.get_image_features(**input).cpu().detach().numpy()[0])
-        return {"clip_embeddings": image_features.cpu().detach().numpy()[0]}
+            print(self.model.get_image_features(**input).cpu().detach().numpy())
+        return {"clip_embeddings": image_features.cpu().detach().numpy()}