Abdul-Ib commited on
Commit
6020ca3
1 Parent(s): 4b47faa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,11 +12,11 @@ from sentence_transformers import SentenceTransformer, util
12
 
13
  # read data
14
  df = pd.read_csv("./assets/final_combined_raw.csv")[['category', 'brand', 'product_name']].to_dict(orient='records')
15
- doc_embeddings = np.load("./assets/multi_embed.npy", allow_pickle=True)
16
 
17
 
18
  # Semantic Search model
19
- semantic_model = SentenceTransformer("Abdul-Ib/paraphrase-multilingual-MiniLM-L12-v2-2024", cache_folder = "./assets")
20
 
21
  # full-text search model
22
  with open('./assets/bm25_L.pkl', 'rb') as bm25result_file:
 
12
 
13
  # read data
14
  df = pd.read_csv("./assets/final_combined_raw.csv")[['category', 'brand', 'product_name']].to_dict(orient='records')
15
+ doc_embeddings = np.load("./assets/e5_embed.npy", allow_pickle=True)
16
 
17
 
18
  # Semantic Search model
19
+ semantic_model = SentenceTransformer("Abdul-Ib/multilingual-e5-small-2024", cache_folder = "./assets")
20
 
21
  # full-text search model
22
  with open('./assets/bm25_L.pkl', 'rb') as bm25result_file: