Nonnormalizable commited on
Commit
9f48354
·
1 Parent(s): 8bae64a

Trying other endpoint structure.

Browse files
Files changed (1) hide show
  1. 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,