wangrongsheng commited on
Commit
88b741a
1 Parent(s): c90545b

support two models

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -126,7 +126,7 @@ def generate(
126
  chat_interface = gr.ChatInterface(
127
  fn=generate,
128
  additional_inputs=[
129
- gr.Dropdown(choices=["INTJ-en", "INTJ-zh"], value="INTJ-en", label="Select Model")
130
  gr.Textbox(label="System prompt", lines=6),
131
  gr.Slider(
132
  label="Max new tokens",
 
126
  chat_interface = gr.ChatInterface(
127
  fn=generate,
128
  additional_inputs=[
129
+ gr.Dropdown(choices=["INTJ-en", "INTJ-zh"], value="INTJ-en", label="Select Model"),
130
  gr.Textbox(label="System prompt", lines=6),
131
  gr.Slider(
132
  label="Max new tokens",