Spaces:
Running
Running
fahmiaziz98
commited on
Commit
·
af36df4
1
Parent(s):
fa16bad
fix error
Browse files- core/embedding.py +2 -2
- core/model_manager.py +0 -1
core/embedding.py
CHANGED
|
@@ -48,8 +48,8 @@ class EmbeddingModel:
|
|
| 48 |
self.load()
|
| 49 |
|
| 50 |
try:
|
| 51 |
-
|
| 52 |
-
return embedding
|
| 53 |
except Exception as e:
|
| 54 |
logger.error(f"Embedding generation failed: {e}")
|
| 55 |
raise
|
|
|
|
| 48 |
self.load()
|
| 49 |
|
| 50 |
try:
|
| 51 |
+
embeddings = self.model.encode_query(text, prompt=prompt)
|
| 52 |
+
return [embedding.tolist() for embedding in embeddings]
|
| 53 |
except Exception as e:
|
| 54 |
logger.error(f"Embedding generation failed: {e}")
|
| 55 |
raise
|
core/model_manager.py
CHANGED
|
@@ -142,7 +142,6 @@ class ModelManager:
|
|
| 142 |
"repository": config.repository,
|
| 143 |
}
|
| 144 |
|
| 145 |
-
|
| 146 |
def generate_api_description(self) -> str:
|
| 147 |
"""Generate a dynamic API description based on available models."""
|
| 148 |
|
|
|
|
| 142 |
"repository": config.repository,
|
| 143 |
}
|
| 144 |
|
|
|
|
| 145 |
def generate_api_description(self) -> str:
|
| 146 |
"""Generate a dynamic API description based on available models."""
|
| 147 |
|