Update main.py

main.py

@@ -28,7 +28,7 @@ EXAMPLE_QUERIES = [
 ]
 EMBEDDING_MODEL_NAME = "allenai-specter"
 LLM_MODEL_NAME = "Qwen/Qwen2-7B-Instruct"
-
+PUBLICATIONS_TO_RETRIEVE = 3
 # Load the dataset and convert to pandas
 data = pandas.read_parquet("hf://datasets/ccm/publications/data/train-00000-of-00001.parquet")
 
@@ -128,7 +128,7 @@ def reply(message: str, history: list[str]) -> str:
     """
 
     # Apply preprocessing
-    message, bypass = preprocess(message, 5)
+    message, bypass = preprocess(message, PUBLICATIONS_TO_RETRIEVE)
 
     # This is some handling that is applied to the history variable to put it in a good format
     history_transformer_format = [