ccm commited on
Commit
0400fe2
·
verified ·
1 Parent(s): 3252ba7

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -23,7 +23,7 @@ EXAMPLE_QUERIES = [
23
  "What can you tell me about analytical target cascading?",
24
  "What is known about different modes for human-AI teaming?",
25
  ]
26
- EMBEDDING_MODEL_NAME = "allenai-specter"
27
  LLM_MODEL_NAME = "Qwen/Qwen2-7B-Instruct"
28
 
29
  # Load the dataset and convert to pandas
 
23
  "What can you tell me about analytical target cascading?",
24
  "What is known about different modes for human-AI teaming?",
25
  ]
26
+ EMBEDDING_MODEL_NAME = "all-MiniLM-L6-v2"
27
  LLM_MODEL_NAME = "Qwen/Qwen2-7B-Instruct"
28
 
29
  # Load the dataset and convert to pandas