Update tasks/text.py
Browse files- 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 = "
|
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":
|
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,
|