Update main.py

main.py

@@ -29,8 +29,10 @@ EXAMPLE_QUERIES = [
 EMBEDDING_MODEL_NAME = "allenai-specter"
 LLM_MODEL_NAME = "Qwen/Qwen2-7B-Instruct"
 PUBLICATIONS_TO_RETRIEVE = 3
+PARQUET_URL = "hf://datasets/ccm/publications/data/train-00000-of-00001.parquet"
+
 # Load the dataset and convert to pandas
-data = pandas.read_parquet("hf://datasets/ccm/publications/data/train-00000-of-00001.parquet")
+data = pandas.read_parquet(PARQUET_URL)
 
 # Filter out any publications without an abstract
 abstract_is_null = [