Query

tasks/text.py

@@ -1,4 +1,4 @@
-from fastapi import APIRouter
+from fastapi import APIRouter, Query
 from datetime import datetime
 from datasets import load_dataset
 from sklearn.metrics import accuracy_score
@@ -73,7 +73,7 @@ def bert_model(test_dataset: dict, model_type: str):
 @router.post(ROUTE, tags=["Text Task"])
 async def evaluate_text(
     request: TextEvaluationRequest,
-    model_type="bert-base",
+    model_type: str = "bert-base",
 ):
     """
     Evaluate text classification for climate disinformation detection.