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):
|