Spaces:
Sleeping
Sleeping
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
|