ccm commited on
Commit
c313486
·
verified ·
1 Parent(s): 4568c7e

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -27,7 +27,7 @@ EXAMPLE_QUERIES = [
27
  "What is known about different modes for human-AI teaming?",
28
  ]
29
  EMBEDDING_MODEL_NAME = "allenai-specter"
30
- LLM_MODEL_NAME = "microsoft/Phi-3-small-8k-instruct" #"Qwen/Qwen2-7B-Instruct"
31
 
32
  # Load the dataset and convert to pandas
33
  data = pandas.read_parquet("hf://datasets/ccm/publications/data/train-00000-of-00001.parquet")
 
27
  "What is known about different modes for human-AI teaming?",
28
  ]
29
  EMBEDDING_MODEL_NAME = "allenai-specter"
30
+ LLM_MODEL_NAME = "Qwen/Qwen2-7B-Instruct"
31
 
32
  # Load the dataset and convert to pandas
33
  data = pandas.read_parquet("hf://datasets/ccm/publications/data/train-00000-of-00001.parquet")