Updated API

tasks/audio.py

@@ -5,8 +5,9 @@ from .utils.emissions import get_space_info
 router = APIRouter()
 
 DESCRIPTION = "Random Baseline"
+ROUTE = "/audio"
 
-@router.post("/audio", tags=["Audio Task"],
+@router.post(ROUTE, tags=["Audio Task"],
              description=DESCRIPTION)
 async def evaluate_audio(request: AudioEvaluationRequest):
     """
@@ -22,6 +23,7 @@ async def evaluate_audio(request: AudioEvaluationRequest):
         "username": username,
         "space_url": space_url,
         "model_description": DESCRIPTION,
+        "route": ROUTE,
         "received_config": {
             "dataset_name": request.dataset_name,
             "test_size": request.test_size,

tasks/image.py

@@ -5,8 +5,9 @@ from .utils.emissions import get_space_info
 router = APIRouter()
 
 DESCRIPTION = "Random Baseline"
+ROUTE = "/image"
 
-@router.post("/image", tags=["Image Task"],
+@router.post(ROUTE, tags=["Image Task"],
              description=DESCRIPTION)
 async def evaluate_image(request: ImageEvaluationRequest):
     """
@@ -21,6 +22,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
         "message": "Image evaluation endpoint not yet implemented",
         "username": username,
         "space_url": space_url,
+        "route": ROUTE,
         "model_description": DESCRIPTION,
         "received_config": {
             "dataset_name": request.dataset_name,

tasks/text.py

@@ -10,8 +10,9 @@ from .utils.emissions import tracker, clean_emissions_data, get_space_info
 router = APIRouter()
 
 DESCRIPTION = "Random Baseline"
+ROUTE = "/text"
 
-@router.post("/text", tags=["Text Task"], 
+@router.post(ROUTE, tags=["Text Task"], 
              description=DESCRIPTION)
 async def evaluate_text(request: TextEvaluationRequest):
     """
@@ -80,6 +81,7 @@ async def evaluate_text(request: TextEvaluationRequest):
         "energy_consumed_wh": emissions_data.energy_consumed * 1000,
         "emissions_gco2eq": emissions_data.emissions * 1000,
         "emissions_data": clean_emissions_data(emissions_data),
+        "api_route": ROUTE,
         "dataset_config": {
             "dataset_name": request.dataset_name,
             "test_size": request.test_size,