submission-template

Sleeping

evgeniiarazum commited on Jan 31

Commit

d753f94

verified ·

1 Parent(s): 088271e

Update tasks/text.py

Files changed (1) hide show

tasks/text.py CHANGED Viewed

@@ -13,7 +13,7 @@ from .utils.emissions import tracker, clean_emissions_data, get_space_info
 router = APIRouter()
-MODEL_TYPE = "mpnet_frugalai"
 DESCRIPTIONS = {
     "distilbert_frugalai": "distilbert tuned on frugal ai data",
     "modernbert_frugalai": "distilbert tuned on frugal ai data",

 router = APIRouter()
+MODEL_TYPE = "baseline"
 DESCRIPTIONS = {
     "distilbert_frugalai": "distilbert tuned on frugal ai data",
     "modernbert_frugalai": "distilbert tuned on frugal ai data",