Pointing at bert-medium

Finetune BERT.ipynb

@@ -472,7 +472,7 @@
     "acc 0.650, energy 0.129 Wh\n",
     "\n",
     "bert-small\\\n",
-    "acc, energy\n",
+    "acc 0.656, energy 0.256 Wh\n",
     "\n",
     "bert-medium\\\n",
     "acc, energy\n",

tasks/text.py

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