aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorShelbySolomonDarnell2024-10-01 18:59:06 +0300
committerShelbySolomonDarnell2024-10-01 18:59:06 +0300
commitc76a305124d92a2d390a7c6d979c1fb9ff7c5f7d (patch)
tree142b76973be6f71fca91c61e701087971011854b
parent2c1e9099d34a0600918cfbe87b32d0a05003b3ef (diff)
downloadgn-ai-c76a305124d92a2d390a7c6d979c1fb9ff7c5f7d.tar.gz
Mod script to get responses
-rw-r--r--gnqa/paper2_eval/src/run_questions.py4
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