Hot potatoes

src/LBP.py

@@ -31,5 +31,6 @@ class LBPImageEncoder:
         # face = np.array(face)
 
         lbp, hist = self.describe(face)
-        return lbp, hist
+
+        return hist
 

src/app.py

@@ -25,7 +25,7 @@ def index_dataset(dataset):
 
     ## LBP Embeddings
     lbp_model = LBPImageEncoder()
-    dataset_with_embeddings = dataset_with_embeddings.map(lambda row: {'lbp_embeddings': lbp_model.describe(row["image"])})
+    dataset_with_embeddings = dataset_with_embeddings.map(lambda row: {'lbp_embeddings': lbp_model.preprocess_img(row["image"])})
 
     # Add index
     dataset_with_embeddings.add_faiss_index(column='color_embeddings')