Add HTTPException for invalid token validation

main.py

@@ -7,9 +7,10 @@ from pathlib import Path
 from typing import Annotated
 
 from dotenv import load_dotenv
-from fastapi import BackgroundTasks, FastAPI, Header
+from fastapi import BackgroundTasks, FastAPI, Header, HTTPException
 from fastapi.responses import JSONResponse
-from huggingface_hub import CommitScheduler, HfApi, hf_hub_download
+from huggingface_hub import CommitScheduler, HfApi, hf_hub_download, whoami
+from huggingface_hub.utils._errors import HTTPError
 from pydantic import BaseModel, Field
 from starlette.responses import RedirectResponse
 
@@ -86,7 +87,11 @@ class Vote(BaseModel):
 
 
 def validate_token(token: str = Header(None)):
-    return token == "ABCD"
+    try:
+        whoami(token)
+        return True
+    except HTTPError:
+        return False
 
 
 @app.post("/vote")
@@ -96,7 +101,7 @@ async def receive_vote(
     background_tasks: BackgroundTasks,
 ):
     if not validate_token(Authorization):
-        return JSONResponse(status_code=401, content={"message": "Invalid token"})
+        raise HTTPException(status_code=401, detail="Invalid token")
     vote_entry = {
         "dataset": vote.dataset,
         "vote": vote.vote,