ccm commited on
Commit
3f75e1c
·
verified ·
1 Parent(s): 8d4ff63

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -23,7 +23,7 @@ EMBEDDING_MODEL_NAME = "allenai-specter"
23
  LLM_MODEL_NAME = "Qwen/Qwen2-7B-Instruct"
24
 
25
  # Load the dataset and convert to pandas
26
- data = pd.read_parquet("hf://datasets/ccm/rag-idetc/data/train-00000-of-00001.parquet")
27
 
28
  # Filter out any publications without an abstract
29
  abstract_is_null = [
 
23
  LLM_MODEL_NAME = "Qwen/Qwen2-7B-Instruct"
24
 
25
  # Load the dataset and convert to pandas
26
+ data = pandas.read_parquet("hf://datasets/ccm/rag-idetc/data/train-00000-of-00001.parquet")
27
 
28
  # Filter out any publications without an abstract
29
  abstract_is_null = [