Spaces:
Sleeping
Sleeping
Commit
·
df03c96
1
Parent(s):
b92d070
bug: fixed warning 'schema_extra' has been renamed to 'json_schema_extra'
Browse files
open/text/embeddings/server/app.py
CHANGED
@@ -42,7 +42,7 @@ class CreateEmbeddingRequest(BaseModel):
|
|
42 |
user: Optional[str]
|
43 |
|
44 |
class Config:
|
45 |
-
|
46 |
"example": {
|
47 |
"input": "The food was delicious and the waiter...",
|
48 |
}
|
@@ -82,12 +82,12 @@ def _create_embedding(
|
|
82 |
encode_kwargs=encode_kwargs)
|
83 |
elif model_name.startswith("BAAI/bge-") and model_name.endswith("-en"):
|
84 |
embeddings = HuggingFaceBgeEmbeddings(model_name=model_name,
|
85 |
-
|
86 |
-
|
87 |
elif model_name.startswith("BAAI/bge-") and model_name.endswith("-zh"):
|
88 |
embeddings = HuggingFaceBgeEmbeddings(model_name=model_name,
|
89 |
-
|
90 |
-
|
91 |
else:
|
92 |
embeddings = HuggingFaceEmbeddings(
|
93 |
model_name=model_name, encode_kwargs=encode_kwargs)
|
|
|
42 |
user: Optional[str]
|
43 |
|
44 |
class Config:
|
45 |
+
json_schema_extra = {
|
46 |
"example": {
|
47 |
"input": "The food was delicious and the waiter...",
|
48 |
}
|
|
|
82 |
encode_kwargs=encode_kwargs)
|
83 |
elif model_name.startswith("BAAI/bge-") and model_name.endswith("-en"):
|
84 |
embeddings = HuggingFaceBgeEmbeddings(model_name=model_name,
|
85 |
+
query_instruction=BGE_EN_QUERY_INSTRUCTION,
|
86 |
+
encode_kwargs=encode_kwargs)
|
87 |
elif model_name.startswith("BAAI/bge-") and model_name.endswith("-zh"):
|
88 |
embeddings = HuggingFaceBgeEmbeddings(model_name=model_name,
|
89 |
+
query_instruction=BGE_ZH_QUERY_INSTRUCTION,
|
90 |
+
encode_kwargs=encode_kwargs)
|
91 |
else:
|
92 |
embeddings = HuggingFaceEmbeddings(
|
93 |
model_name=model_name, encode_kwargs=encode_kwargs)
|