Spaces:
Sleeping
Sleeping
chore(defaults): set BioBERT v1.1 + conll2003 and epochs=3 as baseline
#10
by SHA888 - opened
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.
|
| 12 |
-
DEFAULT_DATASET = "
|
| 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=
|
| 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():
|