Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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[
|
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
|