diff --git a/nlp/normalize.py b/nlp/normalize.py index ae56a4e..4495e90 100644 --- a/nlp/normalize.py +++ b/nlp/normalize.py @@ -26,7 +26,7 @@ def segment_text(text): """ # Encode the input text for the model # inputs = tokenizer.encode("segment: " + text, return_tensors="pt") - inputs = tokenizer("segment: " + input_text, return_tensors="pt") + inputs = tokenizer("segment: " + text, return_tensors="pt") # Generate predictions with torch.no_grad():