normlizer-fix3

2024-09-27 10:16:47 +03:00
parent e1ae8a2de0
commit e2761ee3d8
1 changed files with 1 additions and 1 deletions
--- 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():