danhtran2mind's picture
Upload folder using huggingface_hub
df08b89 verified
import string
def input_processing(in_string):
punct_marks = string.punctuation
for mark in punct_marks:
if mark in in_string:
in_string = in_string.replace(mark, " " + mark)
in_string = in_string.replace("scholl", "school")
in_string = in_string.strip()
return in_string
def output_processing(in_string):
in_string = in_string.replace("_", " ").replace('<sos>', '').replace('<eos>', '').strip()
in_string = in_string[0].upper() + in_string[1:]
return in_string