Pointing at bert-base

Finetune BERT.ipynb

@@ -475,7 +475,7 @@
     "acc 0.656, energy 0.256 Wh\n",
     "\n",
     "bert-medium\\\n",
-    "acc, energy\n",
+    "acc 0.645, energy 0.273 Wh\n",
     "\n",
     "bert-base\\\n",
     "acc, energy"

tasks/text.py

@@ -13,7 +13,7 @@ from .utils.emissions import tracker, clean_emissions_data, get_space_info
 
 router = APIRouter()
 
-MODEL_TYPE = "bert-medium"
+MODEL_TYPE = "bert-base"
 DESCRIPTIONS = {
     "baseline": "baseline most common class",
     "bert-base": "bert base fine tuned on just training data, Nvidia T4 small",