beyoru commited on
Commit
4b6e1a7
·
verified ·
1 Parent(s): 157d4f6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ import re
6
  # Model configuration
7
  MAX_REASONING_TOKENS = 4096
8
  MAX_RESPONSE_TOKENS = 2048
9
- MODEL_OPTIONS = ["beyoru/ThinkAgain1.3", "beyoru/ThinkAgain1.4", "beyoru/ThinkAgain1.5"]
10
 
11
  # Global variables for model and tokenizer
12
  model = None
@@ -68,7 +68,7 @@ with gr.Blocks() as demo:
68
  gr.Markdown("# ThinkAgain Chatbot")
69
 
70
  # Model selection
71
- model_dropdown = gr.Dropdown(choices=MODEL_OPTIONS, label="Select Model", value=MODEL_OPTIONS[0])
72
  model_load_status = gr.Textbox(label="Model Status", interactive=False)
73
 
74
  # Load model when dropdown changes
 
6
  # Model configuration
7
  MAX_REASONING_TOKENS = 4096
8
  MAX_RESPONSE_TOKENS = 2048
9
+ MODEL_OPTIONS = ["beyoru/ThinkAgain1.3", "beyoru/ThinkAgain1.4", "beyoru/ThinkAgain1.5, beyoru/SmallAgent-1-zh"]
10
 
11
  # Global variables for model and tokenizer
12
  model = None
 
68
  gr.Markdown("# ThinkAgain Chatbot")
69
 
70
  # Model selection
71
+ model_dropdown = gr.Dropdown(choices=MODEL_OPTIONS, label="Select Model", value=MODEL_OPTIONS[-1])
72
  model_load_status = gr.Textbox(label="Model Status", interactive=False)
73
 
74
  # Load model when dropdown changes