asaf1602 commited on
Commit
bb8fa77
·
verified ·
1 Parent(s): b8397a5

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. logic/search.py +1 -1
logic/search.py CHANGED
@@ -5,7 +5,7 @@ import faiss
5
  from sentence_transformers import SentenceTransformer, CrossEncoder
6
 
7
  class SloganSearcher:
8
- def _init_(self, assets_dir="assets", use_rerank=False, rerank_model="cross-encoder/stsb-roberta-base"):
9
  meta_path = os.path.join(assets_dir, "meta.json")
10
  if not os.path.exists(meta_path):
11
  raise FileNotFoundError(f"Missing {meta_path}. Build assets first.")
 
5
  from sentence_transformers import SentenceTransformer, CrossEncoder
6
 
7
  class SloganSearcher:
8
+ def __init__(self, assets_dir="assets", use_rerank=False, rerank_model="cross-encoder/stsb-roberta-base"):
9
  meta_path = os.path.join(assets_dir, "meta.json")
10
  if not os.path.exists(meta_path):
11
  raise FileNotFoundError(f"Missing {meta_path}. Build assets first.")