sibthinon commited on
Commit
d64c781
·
verified ·
1 Parent(s): 67f68f8

change model to finetune e5

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -29,8 +29,8 @@ api = Api(AIRTABLE_API_KEY)
29
  table = api.table(BASE_ID, TABLE_NAME)
30
 
31
  # Load model
32
- model = SentenceTransformer('intfloat/multilingual-e5-small')
33
- collection_name = "product_E5"
34
 
35
  # Load whitelist
36
  with open("keyword_whitelist.pkl", "rb") as f:
 
29
  table = api.table(BASE_ID, TABLE_NAME)
30
 
31
  # Load model
32
+ model = SentenceTransformer('e5_finetuned')
33
+ collection_name = "product_E5_finetune"
34
 
35
  # Load whitelist
36
  with open("keyword_whitelist.pkl", "rb") as f: