Spaces:
Sleeping
Sleeping
Update ai_engine.py
Browse files- ai_engine.py +0 -4
ai_engine.py
CHANGED
|
@@ -118,11 +118,7 @@ class LocalModelHandler:
|
|
| 118 |
yield f" [Error generating response: {str(e)}]"
|
| 119 |
|
| 120 |
|
| 121 |
-
# Singleton for embedding model
|
| 122 |
-
_embed_model = None
|
| 123 |
-
|
| 124 |
def get_embedding(text):
|
| 125 |
-
|
| 126 |
return _embed_model.encode(text).tolist()
|
| 127 |
|
| 128 |
def chat_stream(messages):
|
|
|
|
| 118 |
yield f" [Error generating response: {str(e)}]"
|
| 119 |
|
| 120 |
|
|
|
|
|
|
|
|
|
|
| 121 |
def get_embedding(text):
|
|
|
|
| 122 |
return _embed_model.encode(text).tolist()
|
| 123 |
|
| 124 |
def chat_stream(messages):
|