Add support for extensions parameters

Still experimental
This commit is contained in:
oobabooga 2023-01-28 23:00:51 -03:00
parent e779fd795f
commit 6da94e358c
2 changed files with 29 additions and 2 deletions

View file

@ -1,14 +1,18 @@
params = {
"input suffix": " *I say as I make a funny face*",
}
def input_modifier(string):
"""
This function is applied to your text inputs before
they are fed into the model.
"""
return string.replace(' ', '#')
return string + params["input suffix"]
def output_modifier(string):
"""
This function is applied to the model outputs.
"""
return string.replace(' ', '_')
return string