Spaces:
Runtime error
Runtime error
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)
|