Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,10 +9,10 @@ from safetensors.torch import load_file as load_safetensors
|
|
| 9 |
# 🔧 Model versions configuration
|
| 10 |
# ----------------------------
|
| 11 |
MODEL_VERSIONS = {
|
| 12 |
-
"Beeper v3 (
|
| 13 |
"repo_id": "AbstractPhil/beeper-rose-v3",
|
| 14 |
"model_file": "beeper_final.safetensors",
|
| 15 |
-
"description": "Beeper v3 with 30+ epochs including
|
| 16 |
},
|
| 17 |
"Beeper v2 (Extended)": {
|
| 18 |
"repo_id": "AbstractPhil/beeper-rose-v2",
|
|
@@ -139,9 +139,9 @@ def beeper_reply(message, history, model_version, temperature=None, top_k=None,
|
|
| 139 |
temperature=float(temperature) * 0.8, # Slightly lower temp
|
| 140 |
top_k=int(top_k),
|
| 141 |
top_p=float(top_p),
|
| 142 |
-
repetition_penalty=1.
|
| 143 |
presence_penalty=0.8, # Higher presence penalty
|
| 144 |
-
frequency_penalty=0.
|
| 145 |
device=device,
|
| 146 |
detokenize=True
|
| 147 |
)
|
|
@@ -211,12 +211,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
|
| 211 |
with gr.Column(scale=3):
|
| 212 |
model_dropdown = gr.Dropdown(
|
| 213 |
choices=list(MODEL_VERSIONS.keys()),
|
| 214 |
-
value="Beeper
|
| 215 |
label="Select Beeper Version",
|
| 216 |
info="Choose which version of Beeper to chat with"
|
| 217 |
)
|
| 218 |
with gr.Column(scale=7):
|
| 219 |
-
version_info = gr.Markdown("**Current:** Beeper
|
| 220 |
|
| 221 |
# Update version info when dropdown changes
|
| 222 |
def update_version_info(version_name):
|
|
@@ -231,7 +231,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
|
|
| 231 |
|
| 232 |
# Chat interface
|
| 233 |
chatbot = gr.Chatbot(label="Chat with Beeper", type="tuples", height=400)
|
| 234 |
-
msg = gr.Textbox(label="Message", placeholder="Type your message here...")
|
| 235 |
|
| 236 |
with gr.Row():
|
| 237 |
with gr.Column(scale=2):
|
|
|
|
| 9 |
# 🔧 Model versions configuration
|
| 10 |
# ----------------------------
|
| 11 |
MODEL_VERSIONS = {
|
| 12 |
+
"Beeper v3 (Multi-Concept)": {
|
| 13 |
"repo_id": "AbstractPhil/beeper-rose-v3",
|
| 14 |
"model_file": "beeper_final.safetensors",
|
| 15 |
+
"description": "Beeper v3 with 30+ epochs including reasoning, math, coding, and more."
|
| 16 |
},
|
| 17 |
"Beeper v2 (Extended)": {
|
| 18 |
"repo_id": "AbstractPhil/beeper-rose-v2",
|
|
|
|
| 139 |
temperature=float(temperature) * 0.8, # Slightly lower temp
|
| 140 |
top_k=int(top_k),
|
| 141 |
top_p=float(top_p),
|
| 142 |
+
repetition_penalty=1.2, # Higher penalty for repetition
|
| 143 |
presence_penalty=0.8, # Higher presence penalty
|
| 144 |
+
frequency_penalty=0.1, # Add frequency penalty
|
| 145 |
device=device,
|
| 146 |
detokenize=True
|
| 147 |
)
|
|
|
|
| 211 |
with gr.Column(scale=3):
|
| 212 |
model_dropdown = gr.Dropdown(
|
| 213 |
choices=list(MODEL_VERSIONS.keys()),
|
| 214 |
+
value="Beeper v3 (Multi-Concept)",
|
| 215 |
label="Select Beeper Version",
|
| 216 |
info="Choose which version of Beeper to chat with"
|
| 217 |
)
|
| 218 |
with gr.Column(scale=7):
|
| 219 |
+
version_info = gr.Markdown("**Current:** Beeper v3 with 30+ epochs including reasoning, math, coding, and more.")
|
| 220 |
|
| 221 |
# Update version info when dropdown changes
|
| 222 |
def update_version_info(version_name):
|
|
|
|
| 231 |
|
| 232 |
# Chat interface
|
| 233 |
chatbot = gr.Chatbot(label="Chat with Beeper", type="tuples", height=400)
|
| 234 |
+
msg = gr.Textbox(label="Message", placeholder="Type your message here... She will probably complete it for now, but maybe she'll answer.")
|
| 235 |
|
| 236 |
with gr.Row():
|
| 237 |
with gr.Column(scale=2):
|