Update main.py

main.py

@@ -27,8 +27,8 @@ EXAMPLE_QUERIES = [
     "What is known about different modes for human-AI teaming?",
 ]
 EMBEDDING_MODEL_NAME = "allenai-specter"
-LLM_MODEL_NAME = "Qwen/Qwen2-7B-Instruct"
-PUBLICATIONS_TO_RETRIEVE = 3
+LLM_MODEL_NAME = "Qwen/Qwen2.5-7B-Instruct"
+PUBLICATIONS_TO_RETRIEVE = 5
 PARQUET_URL = "hf://datasets/ccm/publications/data/train-00000-of-00001.parquet"
 
 # Load the dataset and convert to pandas