Spaces:
Configuration error
Configuration error
Update cross_encoder_reranking_train.py
Browse files
cross_encoder_reranking_train.py
CHANGED
@@ -134,10 +134,8 @@ def extract_text(content_dict, text_type="full"):
|
|
134 |
for key, value in content_dict.items():
|
135 |
if key.startswith('c-'):
|
136 |
content.append(value)
|
137 |
-
|
138 |
-
|
139 |
-
content.append(value)
|
140 |
-
return " ".join(content)
|
141 |
|
142 |
elif text_type == "tac1":
|
143 |
# Extract title, abstract, and first claim
|
|
|
134 |
for key, value in content_dict.items():
|
135 |
if key.startswith('c-'):
|
136 |
content.append(value)
|
137 |
+
if key == "features":
|
138 |
+
content += list(content_dict[key].values())
|
|
|
|
|
139 |
|
140 |
elif text_type == "tac1":
|
141 |
# Extract title, abstract, and first claim
|