Spaces:
Runtime error
Runtime error
pminervini
commited on
Commit
•
bc4faee
1
Parent(s):
210a90b
update
Browse files
src/backend/tasks/cnndm/task_v2.py
CHANGED
@@ -100,12 +100,12 @@ class CNNDMv2(ConfigurableTask):
|
|
100 |
def test_docs(self):
|
101 |
return self.dataset["test"]
|
102 |
|
103 |
-
def
|
104 |
res = "Provide a summary of the provided article."
|
105 |
return res
|
106 |
|
107 |
def doc_to_text(self, doc):
|
108 |
-
return f'{self.
|
109 |
|
110 |
@staticmethod
|
111 |
def should_decontaminate():
|
|
|
100 |
def test_docs(self):
|
101 |
return self.dataset["test"]
|
102 |
|
103 |
+
def custom_prompt(self):
|
104 |
res = "Provide a summary of the provided article."
|
105 |
return res
|
106 |
|
107 |
def doc_to_text(self, doc):
|
108 |
+
return f'{self.custom_prompt()}\n\nArticle: {doc["article"]}\nSummary:'
|
109 |
|
110 |
@staticmethod
|
111 |
def should_decontaminate():
|