2023-02-24 19:00:11 -03:00
|
|
|
import gradio as gr
|
|
|
|
|
2023-01-28 23:00:51 -03:00
|
|
|
params = {
|
2023-02-26 02:37:21 -03:00
|
|
|
"activate": True,
|
|
|
|
"bias string": " *I am so happy*",
|
2023-01-28 23:00:51 -03:00
|
|
|
}
|
|
|
|
|
2023-01-27 00:40:39 -03:00
|
|
|
def input_modifier(string):
|
|
|
|
"""
|
|
|
|
This function is applied to your text inputs before
|
|
|
|
they are fed into the model.
|
|
|
|
"""
|
|
|
|
|
2023-02-10 15:24:26 -03:00
|
|
|
return string
|
2023-01-27 00:40:39 -03:00
|
|
|
|
|
|
|
def output_modifier(string):
|
|
|
|
"""
|
|
|
|
This function is applied to the model outputs.
|
|
|
|
"""
|
|
|
|
|
2023-01-28 23:00:51 -03:00
|
|
|
return string
|
2023-01-29 10:11:59 -03:00
|
|
|
|
|
|
|
def bot_prefix_modifier(string):
|
|
|
|
"""
|
|
|
|
This function is only applied in chat mode. It modifies
|
|
|
|
the prefix text for the Bot and can be used to bias its
|
|
|
|
behavior.
|
|
|
|
"""
|
|
|
|
|
2023-02-26 02:37:21 -03:00
|
|
|
if params['activate'] == True:
|
|
|
|
return f'{string} {params["bias string"].strip()} '
|
|
|
|
else:
|
|
|
|
return string
|
2023-02-24 19:00:11 -03:00
|
|
|
|
|
|
|
def ui():
|
|
|
|
# Gradio elements
|
2023-02-26 02:37:21 -03:00
|
|
|
activate = gr.Checkbox(value=params['activate'], label='Activate character bias')
|
2023-02-24 19:00:11 -03:00
|
|
|
string = gr.Textbox(value=params["bias string"], label='Character bias')
|
|
|
|
|
|
|
|
# Event functions to update the parameters in the backend
|
|
|
|
string.change(lambda x: params.update({"bias string": x}), string, None)
|
2023-02-26 02:37:21 -03:00
|
|
|
activate.change(lambda x: params.update({"activate": x}), activate, None)
|