Commit
·
9f48354
1
Parent(s):
8bae64a
Trying other endpoint structure.
Browse files- tasks/text.py +10 -0
tasks/text.py
CHANGED
@@ -70,6 +70,16 @@ def bert_model(test_dataset: dict, model_type: str):
|
|
70 |
return predictions
|
71 |
|
72 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
73 |
@router.post(ROUTE, tags=["Text Task"])
|
74 |
async def evaluate_text(
|
75 |
request: TextEvaluationRequest,
|
|
|
70 |
return predictions
|
71 |
|
72 |
|
73 |
+
@router.post("/text-bert-base", tags=["Text Task"])
|
74 |
+
async def evauate_text_model_1(request: TextEvaluationRequest):
|
75 |
+
return evaluate_text(request, model_type="bert-base")
|
76 |
+
|
77 |
+
|
78 |
+
@router.post("/text-baseline", tags=["Text Task"])
|
79 |
+
async def evauate_text_model_2(request: TextEvaluationRequest):
|
80 |
+
return evaluate_text(request, model_type="baseline")
|
81 |
+
|
82 |
+
|
83 |
@router.post(ROUTE, tags=["Text Task"])
|
84 |
async def evaluate_text(
|
85 |
request: TextEvaluationRequest,
|