Spaces:
Sleeping
Sleeping
Python version 3.11
Browse files
README.md
CHANGED
|
@@ -9,6 +9,7 @@ app_file: app.py
|
|
| 9 |
pinned: false
|
| 10 |
license: mit
|
| 11 |
short_description: Chatbot assistant for the CAMELS simulations documentation
|
|
|
|
| 12 |
---
|
| 13 |
|
| 14 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
|
|
|
| 9 |
pinned: false
|
| 10 |
license: mit
|
| 11 |
short_description: Chatbot assistant for the CAMELS simulations documentation
|
| 12 |
+
python_version: 3.11
|
| 13 |
---
|
| 14 |
|
| 15 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
worker.py
CHANGED
|
@@ -49,7 +49,6 @@ def init_llm():
|
|
| 49 |
#Initialize embeddings using a pre-trained model to represent the text data.
|
| 50 |
embedddings_model = "sentence-transformers/multi-qa-distilbert-cos-v1"
|
| 51 |
# embedddings_model = "sentence-transformers/all-MiniLM-L6-v2"
|
| 52 |
-
embedddings_model = "nvidia/NV-Embed-v2"
|
| 53 |
embeddings = HuggingFaceInstructEmbeddings(
|
| 54 |
model_name=embedddings_model,
|
| 55 |
model_kwargs={"device": DEVICE}
|
|
|
|
| 49 |
#Initialize embeddings using a pre-trained model to represent the text data.
|
| 50 |
embedddings_model = "sentence-transformers/multi-qa-distilbert-cos-v1"
|
| 51 |
# embedddings_model = "sentence-transformers/all-MiniLM-L6-v2"
|
|
|
|
| 52 |
embeddings = HuggingFaceInstructEmbeddings(
|
| 53 |
model_name=embedddings_model,
|
| 54 |
model_kwargs={"device": DEVICE}
|