chahah commited on
Commit
3b774af
·
verified ·
1 Parent(s): c64b51b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -82,8 +82,9 @@ def RAG(llm, docs, embeddings):
82
  llm = ChatMistralAI(model="mistral-large-latest", rate_limiter=rate_limiter)
83
 
84
  # Embeddings
85
- embed_model = "sentence-transformers/multi-qa-distilbert-cos-v1"
86
  # embed_model = "nvidia/NV-Embed-v2"
 
87
  embeddings = HuggingFaceInstructEmbeddings(model_name=embed_model)
88
  # embeddings = MistralAIEmbeddings()
89
 
 
82
  llm = ChatMistralAI(model="mistral-large-latest", rate_limiter=rate_limiter)
83
 
84
  # Embeddings
85
+ #embed_model = "sentence-transformers/multi-qa-distilbert-cos-v1"
86
  # embed_model = "nvidia/NV-Embed-v2"
87
+ embed_model = "allenai/specter2"
88
  embeddings = HuggingFaceInstructEmbeddings(model_name=embed_model)
89
  # embeddings = MistralAIEmbeddings()
90