From 44ee7a6e980fd48ea4216e78715d299151002503 Mon Sep 17 00:00:00 2001
From: Dipam Chakraborty <dipam@aicrowd.com>
Date: Sun, 7 Apr 2024 15:06:26 +0000
Subject: [PATCH] Update local_evaluation.py
---
local_evaluation.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/local_evaluation.py b/local_evaluation.py
index 3f12a9d..4dafefe 100644
--- a/local_evaluation.py
+++ b/local_evaluation.py
@@ -92,7 +92,7 @@ def generate_predictions(dataset_path, participant_model):
for query_dict, query_web_search_results in tqdm(zip(qa, web_results), total=len(qa), desc="Generating Predictions"):
query = query_dict["query"]
prediction = participant_model.generate_answer(
- query, query_web_search_results
+ query, query_web_search_results['search_response']
)
# trim prediction to 75 tokens
prediction = trim_predictions_to_max_token_length(prediction)
--
GitLab