PabloVD commited on
Commit
192d160
·
1 Parent(s): 25c6d42

Python version 3.11

Browse files
Files changed (2) hide show
  1. README.md +1 -0
  2. worker.py +0 -1
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}