From c82574db7d5364b9a45477832e30dbb9978dfa98 Mon Sep 17 00:00:00 2001 From: Zaid Alyafeai Date: Sat, 22 Jun 2024 00:21:58 +0000 Subject: [PATCH] reuse the abstract --- utils/embeddings_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/embeddings_utils.py b/utils/embeddings_utils.py index efb150d..09edd65 100644 --- a/utils/embeddings_utils.py +++ b/utils/embeddings_utils.py @@ -62,4 +62,4 @@ def compute_embeddings(texts: List[str], model_name: str) -> List[List[float]]: def build_dataset_prompt(dataset: Dict[str, Union[str, int]]) -> str: - return f"{dataset['Name']} {dataset['Description']} {dataset['Tasks']}" + return f"{dataset['Name']} {dataset['Description']} {dataset['Tasks']} {dataset['Abstract']}"