HonestAnnie commited on
Commit
4295f9e
1 Parent(s): ff486ce
Files changed (2) hide show
  1. app.py +1 -1
  2. chroma/chroma.sqlite3 +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ from spaces import GPU
11
 
12
  @GPU
13
  def get_embeddings(text, task):
14
- model = SentenceTransformer("Linq-AI-Research/Linq-Embed-Mistral")
15
  task = "Given a question, retrieve passages that answer the question"
16
  prompt = f"Instruct: {task}\nQuery: {text}" # Use text here
17
  query_embeddings = model.encode([prompt], convert_to_tensor=True) # Ensure it's a list
 
11
 
12
  @GPU
13
  def get_embeddings(text, task):
14
+ model = SentenceTransformer("Linq-AI-Research/Linq-Embed-Mistral", use_auth_token=os.getenv("HF_TOKEN"))
15
  task = "Given a question, retrieve passages that answer the question"
16
  prompt = f"Instruct: {task}\nQuery: {text}" # Use text here
17
  query_embeddings = model.encode([prompt], convert_to_tensor=True) # Ensure it's a list
chroma/chroma.sqlite3 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6509430e2a2bf98572a286bc10efec77a5edd30505334c2680c06e4b9f1c7968
3
  size 235212800
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d6a9b16016687d388396d462b8e417318533839195eec10bc6b2c0c8f44e1dd8
3
  size 235212800