From 0c76c584018ce7ee1f1631346fa9012d5507788e Mon Sep 17 00:00:00 2001
From: friebolin <friebolin@cl.uni-heidelberg.de>
Date: Fri, 24 Feb 2023 15:37:13 +0100
Subject: [PATCH] Fix split sentence
---
inference.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/inference.py b/inference.py
index 2338d33..313bfad 100644
--- a/inference.py
+++ b/inference.py
@@ -65,7 +65,7 @@ label = int(input())
# Convert to data sample for BERT
-data_sample = [{"sentence": sentence, "pos": pos, "label": label}]
+data_sample = [{"sentence": split_sentence, "pos": pos, "label": label}]
#print(data_sample)
tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
--
GitLab