errors with punctuation

[sreckojovancevic]

def add_punctuation(text):
"""Dodaje interpunkciju pomoću NLP modela i osigurava ispravan format."""
try:
result = punc_model.restore_punctuation(text)
except Exception as e:
print(f"Error in restore_punctuation: {e}")
return text # Return the original text if there's an error

allowed_punctuation = {".", ",", "?", "-", ":"}  # Dozvoljeni znakovi
corrected_text = []

for token in result.split():
    token = str(token)  # Konvertuj sve u string
    if token.isdigit():  # Ako je broj, preskači ga
        continue
    if token in allowed_punctuation or token.isalpha():  # Dozvoljeni znaci i reči
        corrected_text.append(token)

return " ".join(corrected_text)

throws thiss error it giver 500 on web
Traceback (most recent call last):
File "/home/srecko/ATT/vosk_env/lib/python3.12/site-packages/flask/app.py", line 1536, in call
return self.wsgi_app(environ, start_response)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/srecko/ATT/vosk_env/lib/python3.12/site-packages/flask/app.py", line 1514, in wsgi_app
response = self.handle_exception(e)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/srecko/ATT/vosk_env/lib/python3.12/site-packages/flask/app.py", line 1511, in wsgi_app
response = self.full_dispatch_request()
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/srecko/ATT/vosk_env/lib/python3.12/site-packages/flask/app.py", line 919, in full_dispatch_request
rv = self.handle_user_exception(e)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/srecko/ATT/vosk_env/lib/python3.12/site-packages/flask/app.py", line 917, in full_dispatch_request
rv = self.dispatch_request()
^^^^^^^^^^^^^^^^^^^^^^^
File "/home/srecko/ATT/vosk_env/lib/python3.12/site-packages/flask/app.py", line 902, in dispatch_request
return self.ensure_sync(self.view_functions[rule.endpoint])(**view_args) # type: ignore[no-any-return]
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/srecko/ATT/app.py", line 79, in transcribe
transcription_text = add_punctuation(transcription_text)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/srecko/ATT/app.py", line 31, in add_punctuation
result = punc_model.restore_punctuation(text)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/srecko/ATT/vosk_env/lib/python3.12/site-packages/deepmultilingualpunctuation/punctuationmodel.py", line 22, in restore_punctuation
return self.prediction_to_text(result)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/srecko/ATT/vosk_env/lib/python3.12/site-packages/deepmultilingualpunctuation/punctuationmodel.py", line 73, in prediction_to_text
if label in ".,?-:":
^^^^^^^^^^^^^^^^^
TypeError: 'in ' requires string as left operand, not int