Update main.py

main.py

@@ -23,8 +23,7 @@ EXAMPLE_QUERIES = [
     "What can agent-based models do about climate change?",
 ]
 EMBEDDING_MODEL_NAME = "allenai-specter"
-LLM_MODEL_NAME = "Qwen/Qwen2-7B-Instruct"
-# LLM_MODEL_NAME = "Qwen/Qwen2-0.5B-Instruct"
+LLM_MODEL_NAME = "Qwen/Qwen2-1.5B-Instruct"
 
 # Load the dataset and convert to pandas
 data = datasets.load_dataset("ccm/publications")["train"].to_pandas()