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}
|