gizemsarsinlar commited on
Commit
78e545c
1 Parent(s): 9272309

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,9 +32,9 @@ available_llms = ChatNVIDIA.get_available_models(api_key=api_key)
32
  print("Available NVIDIA Language Models:", available_llms)
33
 
34
  # NVIDIAEmbeddings.get_available_models()
35
- embedder = NVIDIAEmbeddings(model="nvidia/embed-qa-4", api_key=api_key, truncate="NONE")
36
  # ChatNVIDIA.get_available_models()
37
- instruct_llm = ChatNVIDIA(model="mistralai/mixtral-8x7b-instruct-v0.1")
38
 
39
  embed_dims = len(embedder.embed_query("test"))
40
  def default_FAISS():
 
32
  print("Available NVIDIA Language Models:", available_llms)
33
 
34
  # NVIDIAEmbeddings.get_available_models()
35
+ embedder = NVIDIAEmbeddings(model="NV-Embed-QA", api_key=api_key, truncate="END")
36
  # ChatNVIDIA.get_available_models()
37
+ instruct_llm = ChatNVIDIA(model="mistralai/mixtral-8x7b-instruct-v0.1", api_key=api_key)
38
 
39
  embed_dims = len(embedder.embed_query("test"))
40
  def default_FAISS():