AbstractPhil commited on
Commit
48dfa68
·
verified ·
1 Parent(s): 4360288

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
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 (Philosophy)": {
13
  "repo_id": "AbstractPhil/beeper-rose-v3",
14
  "model_file": "beeper_final.safetensors",
15
- "description": "Beeper v3 with 30+ epochs including ethics & philosophy"
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.3, # Higher penalty for repetition
143
  presence_penalty=0.8, # Higher presence penalty
144
- frequency_penalty=0.2, # Add frequency penalty
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 v1 (Original)",
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 v1 - Original training on TinyStories")
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):