瀏覽代碼

Add checkbox to disable the character bias

oobabooga 2 年之前
父節點
當前提交
3c9432f039
共有 1 個文件被更改,包括 8 次插入2 次删除
  1. 8 2
      extensions/character_bias/script.py

+ 8 - 2
extensions/character_bias/script.py

@@ -1,7 +1,8 @@
 import gradio as gr
 
 params = {
-    "bias string": " *I speak in an annoyingly cute way*",
+    "activate": True,
+    "bias string": " *I am so happy*",
 }
 
 def input_modifier(string):
@@ -26,11 +27,16 @@ def bot_prefix_modifier(string):
     behavior.
     """
 
-    return f'{string} {params["bias string"].strip()} '
+    if params['activate'] == True:
+        return f'{string} {params["bias string"].strip()} '
+    else:
+        return string
 
 def ui():
     # Gradio elements
+    activate = gr.Checkbox(value=params['activate'], label='Activate character bias')
     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)
+    activate.change(lambda x: params.update({"activate": x}), activate, None)