Samuel Schmidt commited on
Commit
c4eb027
·
1 Parent(s): 1b4e6a4

Update src/app.py

Browse files
Files changed (1) hide show
  1. src/app.py +2 -2
src/app.py CHANGED
@@ -22,7 +22,7 @@ def index_dataset(dataset):
22
 
23
  print("LBP Embeddings")
24
  lbp_model = LBPImageEncoder(8,2)
25
- dataset_with_embeddings = dataset_with_embeddings.map(lambda row: {'lbp_embeddings': lbp_model.preprocess_img(row["image"])})
26
 
27
  # Add index
28
  dataset_with_embeddings.add_faiss_index(column='color_embeddings')
@@ -78,7 +78,7 @@ def get_neighbors(query_image, selected_descriptor, top_k=5):
78
  return images
79
  if "LBP" in selected_descriptor:
80
  lbp_model = LBPImageEncoder(8,2)
81
- qi_embedding = lbp_model.preprocess_img(query_image)
82
  scores, retrieved_examples = dataset_with_embeddings.get_nearest_examples(
83
  'lbp_embeddings', qi_embedding, k=top_k)
84
  images = retrieved_examples['image']
 
22
 
23
  print("LBP Embeddings")
24
  lbp_model = LBPImageEncoder(8,2)
25
+ dataset_with_embeddings = dataset_with_embeddings.map(lambda row: {'lbp_embeddings': lbp_model.describe(row["image"])})
26
 
27
  # Add index
28
  dataset_with_embeddings.add_faiss_index(column='color_embeddings')
 
78
  return images
79
  if "LBP" in selected_descriptor:
80
  lbp_model = LBPImageEncoder(8,2)
81
+ qi_embedding = lbp_model.describe(query_image)
82
  scores, retrieved_examples = dataset_with_embeddings.get_nearest_examples(
83
  'lbp_embeddings', qi_embedding, k=top_k)
84
  images = retrieved_examples['image']