evgeniiarazum commited on
Commit
a311ff6
·
verified ·
1 Parent(s): 29aa520

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. tasks/text.py +3 -2
tasks/text.py CHANGED
@@ -13,9 +13,10 @@ from .utils.emissions import tracker, clean_emissions_data, get_space_info
13
 
14
  router = APIRouter()
15
 
16
- MODEL_TYPE = "distilbert_frugalai"
17
  DESCRIPTIONS = {
18
- "distilbert_frugalai": "distilbert tuned on frugal ai data"
 
19
  }
20
  ROUTE = "/text"
21
 
 
13
 
14
  router = APIRouter()
15
 
16
+ MODEL_TYPE = "modernbert_frugalai"
17
  DESCRIPTIONS = {
18
+ "distilbert_frugalai": "distilbert tuned on frugal ai data",
19
+ "modernbert_frugalai": "distilbert tuned on frugal ai data"
20
  }
21
  ROUTE = "/text"
22