Merge branch 'embeddings-ui' of https://github.com/blakeblackshear/fr…

…igate into embeddings-ui
blakeblackshear · Oct 16, 2024 · a894cd6 · a894cd6
2 parents 6664ec6 + a1abaed
commit a894cd6
Show file tree

Hide file tree

Showing 4 changed files with 262 additions and 578 deletions.
diff --git a/frigate/embeddings/embeddings.py b/frigate/embeddings/embeddings.py
@@ -178,7 +178,7 @@ def batch_upsert_description(self, event_descriptions: dict[str, str]) -> ndarra
         embeddings = []
 
         for desc in event_descriptions.values():
-            embeddings.append(self.text_embedding([desc]))
+            embeddings.append(self.text_embedding([desc])[0])
 
         ids = list(event_descriptions.keys())