Pointing at bert-mini

Finetune BERT.ipynb

@@ -463,9 +463,8 @@
     "[bert-base some hp tuning](https://huggingface.co/datasets/frugal-ai-challenge/public-leaderboard-text/blob/main/submissions/Nonnormalizable_20250120_231350.json):\\\n",
     "acc 0.707, energy 0.803 Wh\n",
     "\n",
-    "bert-tiny, Nvidia 1xL40S:\n",
-    "\n",
-    "bert-tiny, "
+    "bert-tiny, Nvidia T4 small:\\\n",
+    "acc 0.618, energy 0.079 Wh"
    ]
   },
   {

tasks/text.py

@@ -13,6 +13,7 @@ from .utils.emissions import tracker, clean_emissions_data, get_space_info
 
 router = APIRouter()
 
+MODEL_TYPE = "bert-mini"
 DESCRIPTIONS = {
     "baseline": "baseline most common class",
     "bert-base": "bert base fine tuned on just training data, Nvidia T4 small",
@@ -21,7 +22,6 @@ DESCRIPTIONS = {
     "bert-mini": "bert min fine tuned on just training data, Nvidia T4 small",
     "bert-tiny": "bert tiny fine tuned on just training data, Nvidia T4 small",
 }
-
 ROUTE = "/text"
 
 
@@ -92,7 +92,7 @@ def bert_model(test_dataset: dict, model_type: str):
 @router.post(ROUTE, tags=["Text Task"])
 async def evaluate_text(
     request: TextEvaluationRequest,
-    model_type: str = "bert-tiny",
+    model_type: str = MODEL_TYPE,
     # This should be an API query parameter, but it looks like the submission repo
     # https://huggingface.co/spaces/frugal-ai-challenge/submission-portal
     # is built in a way to not accept any other endpoints or parameters.