Pclanglais commited on
Commit
feacba8
1 Parent(s): 27f6f13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -12,6 +12,7 @@ import pandas as pd
12
  from chromadb.config import Settings
13
  from chromadb.utils import embedding_functions
14
  from FlagEmbedding import BGEM3FlagModel
 
15
 
16
  model = BGEM3FlagModel('BAAI/bge-m3',
17
  use_fp16=True) # Setting use_fp16 to True speeds up computation with a slight performance degradation
 
12
  from chromadb.config import Settings
13
  from chromadb.utils import embedding_functions
14
  from FlagEmbedding import BGEM3FlagModel
15
+ from sklearn.metrics.pairwise import cosine_similarity
16
 
17
  model = BGEM3FlagModel('BAAI/bge-m3',
18
  use_fp16=True) # Setting use_fp16 to True speeds up computation with a slight performance degradation