Spaces:
Sleeping
Sleeping
Update src/preprocess/embedding.py
Browse files
src/preprocess/embedding.py
CHANGED
|
@@ -3,7 +3,8 @@ from utils import bt_embeddings
|
|
| 3 |
from tqdm import tqdm
|
| 4 |
from typing import List
|
| 5 |
from langchain_core.embeddings import Embeddings
|
| 6 |
-
from langchain_core.pydantic_v1 import BaseModel
|
|
|
|
| 7 |
|
| 8 |
class BridgeTowerEmbeddings(BaseModel,Embeddings):
|
| 9 |
""" BridgeTower embedding model """
|
|
|
|
| 3 |
from tqdm import tqdm
|
| 4 |
from typing import List
|
| 5 |
from langchain_core.embeddings import Embeddings
|
| 6 |
+
#from langchain_core.pydantic_v1 import BaseModel
|
| 7 |
+
from pydantic.v1 import BaseModel
|
| 8 |
|
| 9 |
class BridgeTowerEmbeddings(BaseModel,Embeddings):
|
| 10 |
""" BridgeTower embedding model """
|