ccm commited on
Commit
d227ba0
·
verified ·
1 Parent(s): 95be0fc

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -2
main.py CHANGED
@@ -18,8 +18,7 @@ GREETING = (
18
  "Design Research Collective. And the best part is that I always cite my sources! What can I tell you about today?"
19
  )
20
  EMBEDDING_MODEL_NAME = "allenai-specter"
21
- # LLM_MODEL_NAME = "Qwen/Qwen2-7B-Instruct"
22
- LLM_MODEL_NAME = "Qwen/Qwen2-0.5B-Instruct"
23
 
24
  # Load the dataset and convert to pandas
25
  full_data = datasets.load_dataset("ccm/publications")["train"].to_pandas()
 
18
  "Design Research Collective. And the best part is that I always cite my sources! What can I tell you about today?"
19
  )
20
  EMBEDDING_MODEL_NAME = "allenai-specter"
21
+ LLM_MODEL_NAME = "Qwen/Qwen2-7B-Instruct"
 
22
 
23
  # Load the dataset and convert to pandas
24
  full_data = datasets.load_dataset("ccm/publications")["train"].to_pandas()