Oriaz commited on
Commit
bb3ba6b
·
verified ·
1 Parent(s): a2de25b

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. tasks/text.py +2 -2
tasks/text.py CHANGED
@@ -20,7 +20,7 @@ import skops.io as sio
20
 
21
  router = APIRouter()
22
 
23
- DESCRIPTION = "Random test"
24
  ROUTE = "/text"
25
 
26
  @router.post(ROUTE, tags=["Text Task"],
@@ -98,7 +98,7 @@ async def evaluate_text(request: TextEvaluationRequest):
98
  "username": username,
99
  "space_url": space_url,
100
  "submission_timestamp": datetime.now().isoformat(),
101
- "model_description": "API test",
102
  "accuracy": float(accuracy),
103
  "energy_consumed_wh": emissions_data.energy_consumed * 1000,
104
  "emissions_gco2eq": emissions_data.emissions * 1000,
 
20
 
21
  router = APIRouter()
22
 
23
+ DESCRIPTION = "Embedding + Logistic Regression"
24
  ROUTE = "/text"
25
 
26
  @router.post(ROUTE, tags=["Text Task"],
 
98
  "username": username,
99
  "space_url": space_url,
100
  "submission_timestamp": datetime.now().isoformat(),
101
+ "model_description": DESCRIPTION,
102
  "accuracy": float(accuracy),
103
  "energy_consumed_wh": emissions_data.energy_consumed * 1000,
104
  "emissions_gco2eq": emissions_data.emissions * 1000,