Spaces:
Paused
Paused
Commit
•
6ca460a
1
Parent(s):
6d8e1bf
Add token parameter to CommitScheduler constructor and update authorization response
Browse files
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 |
-
|
|
|
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,
|