davanstrien HF staff commited on
Commit
6ca460a
1 Parent(s): 6d8e1bf

Add token parameter to CommitScheduler constructor and update authorization response

Browse files
Files changed (1) hide show
  1. main.py +3 -2
main.py CHANGED
@@ -24,6 +24,7 @@ scheduler = CommitScheduler(
24
  folder_path="votes",
25
  path_in_repo="data",
26
  every=1,
 
27
  hf_api=hf_api,
28
  )
29
 
@@ -31,7 +32,6 @@ scheduler = CommitScheduler(
31
  @asynccontextmanager
32
  async def lifespan(app: FastAPI):
33
  Path("votes").mkdir(exist_ok=True)
34
-
35
  yield
36
 
37
 
@@ -89,7 +89,8 @@ async def receive_vote(
89
  ):
90
  token = authorization.split(" ")[1]
91
  if not validate_token(token):
92
- return {"message": "Invalid user token"}
 
93
  vote_entry = {
94
  "dataset": vote.dataset,
95
  "vote": vote.vote,
 
24
  folder_path="votes",
25
  path_in_repo="data",
26
  every=1,
27
+ token=HF_TOKEN,
28
  hf_api=hf_api,
29
  )
30
 
 
32
  @asynccontextmanager
33
  async def lifespan(app: FastAPI):
34
  Path("votes").mkdir(exist_ok=True)
 
35
  yield
36
 
37
 
 
89
  ):
90
  token = authorization.split(" ")[1]
91
  if not validate_token(token):
92
+ # Return a not authorized response
93
+ return {"message": "Not authorized"}
94
  vote_entry = {
95
  "dataset": vote.dataset,
96
  "vote": vote.vote,