aho/welcomecenterbot
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

normlizer-fix3

Browse Source
This commit is contained in:
Untone 2024-09-27 10:16:47 +03:00
parent e1ae8a2de0
commit e2761ee3d8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nlp/normalize.py
View File

@ -26,7 +26,7 @@ def segment_text(text):
""" """
# Encode the input text for the model # Encode the input text for the model
# inputs = tokenizer.encode("segment: " + text, return_tensors="pt") # 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 # Generate predictions
with torch.no_grad(): with torch.no_grad():