Spaces:
Sleeping
Sleeping
Pclanglais
commited on
Commit
•
feacba8
1
Parent(s):
27f6f13
Update app.py
Browse files
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
|