Spaces:
Runtime error
Runtime error
fixes
Browse files- src/models/model.py +2 -2
src/models/model.py
CHANGED
@@ -76,8 +76,8 @@ class DataModule(Dataset):
|
|
76 |
] = -100
|
77 |
|
78 |
return dict(
|
79 |
-
|
80 |
-
|
81 |
keywords_input_ids=input_encoding["input_ids"].flatten(),
|
82 |
keywords_attention_mask=input_encoding["attention_mask"].flatten(),
|
83 |
labels=labels.flatten(),
|
|
|
76 |
] = -100
|
77 |
|
78 |
return dict(
|
79 |
+
keywords=data_row["input_text"],
|
80 |
+
text=data_row["output_text"],
|
81 |
keywords_input_ids=input_encoding["input_ids"].flatten(),
|
82 |
keywords_attention_mask=input_encoding["attention_mask"].flatten(),
|
83 |
labels=labels.flatten(),
|