chore(defaults): set BioBERT v1.1 + conll2003 and epochs=3 as baseline

#10
by SHA888 - opened
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,8 +8,8 @@ import gradio as gr
8
  from huggingface_hub import HfApi, create_repo, hf_hub_url
9
 
10
 
11
- DEFAULT_BASE_MODEL = "dmis-lab/biobert-base-cased-v1.2"
12
- DEFAULT_DATASET = "wikiann:en" # robust default; medical sets may require custom preprocessing
13
  TARGET_REPO = os.getenv("MEDVLLM_TARGET_REPO", "Junaidi-AI/med-vllm")
14
 
15
 
@@ -368,7 +368,7 @@ def build_ui():
368
  label="Dataset (token classification)",
369
  )
370
  with gr.Row():
371
- epochs = gr.Slider(minimum=1, maximum=3, step=1, value=1, label="Epochs")
372
  batch = gr.Slider(minimum=4, maximum=16, step=2, value=8, label="Batch size")
373
  lr = gr.Textbox(value="2e-5", label="Learning rate")
374
  with gr.Row():
 
8
  from huggingface_hub import HfApi, create_repo, hf_hub_url
9
 
10
 
11
+ DEFAULT_BASE_MODEL = "dmis-lab/biobert-base-cased-v1.1"
12
+ DEFAULT_DATASET = "conll2003" # stronger baseline default
13
  TARGET_REPO = os.getenv("MEDVLLM_TARGET_REPO", "Junaidi-AI/med-vllm")
14
 
15
 
 
368
  label="Dataset (token classification)",
369
  )
370
  with gr.Row():
371
+ epochs = gr.Slider(minimum=1, maximum=5, step=1, value=3, label="Epochs")
372
  batch = gr.Slider(minimum=4, maximum=16, step=2, value=8, label="Batch size")
373
  lr = gr.Textbox(value="2e-5", label="Learning rate")
374
  with gr.Row():