diff options
-rw-r--r-- | gnqa/paper2_eval/src/run_questions.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/gnqa/paper2_eval/src/run_questions.py b/gnqa/paper2_eval/src/run_questions.py index 8805329..053d758 100644 --- a/gnqa/paper2_eval/src/run_questions.py +++ b/gnqa/paper2_eval/src/run_questions.py @@ -2,7 +2,7 @@ import json import sys import os -from r2r import R2R, Document, GenerationConfig, R2RClient +from r2r import R2RClient from document_operations import DocOps, QuestionList ''' @@ -30,7 +30,7 @@ except: qLst = QuestionList(read_file, 1) # second parameter is for verbose output ndx = 1 -for question in qLst.get("domainexpert","aging"): +for question in qLst.get("citizenscientist","aging"): print('Getting response for the following question --> {0}'.format(question)) rag_response[str(ndx)] = client.rag(question) ndx += 1 |