Update main.py

main.py

@@ -3,7 +3,7 @@ import pandas as pd
 from fastapi import FastAPI, HTTPException, Body
 from fastapi.responses import FileResponse
 from pydantic import BaseModel, Field
-from typing import List, Dict, Any, Optional
+from typing import List, Dict, Any, Optional, Union
 from datasets import load_dataset, Dataset, DatasetDict
 from huggingface_hub import HfApi, hf_hub_download
 from datetime import datetime, timezone
@@ -170,7 +170,7 @@ class ErrorResponse(BaseModel):
 
 class AnswerItem(BaseModel):
     task_id: str
-    submitted_answer: str = Field(..., description="The agent's answer for the task_id")
+    submitted_answer: Union[str, int, float]  = Field(..., description="The agent's answer for the task_id. Accepts str, int and float")
 
 class Submission(BaseModel):
     username: str = Field(..., description="Hugging Face username", min_length=1)