qgyd2021 commited on
Commit
72ad2e5
1 Parent(s): c64dba3

[update]add main

Browse files
Files changed (1) hide show
  1. main.py +10 -9
main.py CHANGED
@@ -198,15 +198,16 @@ def main():
198
  history_max_len = gr.Slider(minimum=0, maximum=4096, value=1024, step=1, label="history_max_len")
199
 
200
  with gr.Row():
201
- model_name = gr.Dropdown(
202
- choices=[
203
- "Qwen/Qwen-7B-Chat",
204
- "THUDM/chatglm2-6b",
205
- "baichuan-inc/Baichuan2-7B-Chat",
206
- ],
207
- value="Qwen/Qwen-7B-Chat",
208
- label="model_name",
209
- )
 
210
  gr.Examples(examples=["你好"], inputs=text_box)
211
 
212
  inputs = [
 
198
  history_max_len = gr.Slider(minimum=0, maximum=4096, value=1024, step=1, label="history_max_len")
199
 
200
  with gr.Row():
201
+ with gr.Column(scale=1):
202
+ model_name = gr.Dropdown(
203
+ choices=[
204
+ "Qwen/Qwen-7B-Chat",
205
+ "THUDM/chatglm2-6b",
206
+ "baichuan-inc/Baichuan2-7B-Chat",
207
+ ],
208
+ value="Qwen/Qwen-7B-Chat",
209
+ label="model_name",
210
+ )
211
  gr.Examples(examples=["你好"], inputs=text_box)
212
 
213
  inputs = [