Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Clémentine
commited on
Commit
•
bf23d2b
1
Parent(s):
b4ba8b7
update timer
Browse files
app.py
CHANGED
@@ -1,6 +1,7 @@
|
|
1 |
import os
|
2 |
import logging
|
3 |
import time
|
|
|
4 |
import gradio as gr
|
5 |
import datasets
|
6 |
from huggingface_hub import snapshot_download, WebhooksServer, WebhookPayload, RepoCard
|
@@ -356,10 +357,16 @@ async def update_leaderboard(payload: WebhookPayload) -> None:
|
|
356 |
verification_mode="no_checks"
|
357 |
)
|
358 |
|
|
|
359 |
@webhooks_server.add_webhook
|
360 |
async def update_queue(payload: WebhookPayload) -> None:
|
361 |
"""Redownloads the queue dataset each time it updates"""
|
362 |
if payload.repo.type == "dataset" and payload.event.action == "update":
|
363 |
-
|
|
|
|
|
|
|
|
|
|
|
364 |
|
365 |
webhooks_server.launch()
|
|
|
1 |
import os
|
2 |
import logging
|
3 |
import time
|
4 |
+
import datetime
|
5 |
import gradio as gr
|
6 |
import datasets
|
7 |
from huggingface_hub import snapshot_download, WebhooksServer, WebhookPayload, RepoCard
|
|
|
357 |
verification_mode="no_checks"
|
358 |
)
|
359 |
|
360 |
+
LAST_UPDATE_QUEUE = datetime.datetime.now()
|
361 |
@webhooks_server.add_webhook
|
362 |
async def update_queue(payload: WebhookPayload) -> None:
|
363 |
"""Redownloads the queue dataset each time it updates"""
|
364 |
if payload.repo.type == "dataset" and payload.event.action == "update":
|
365 |
+
current_time = datetime.datetime.now()
|
366 |
+
if current_time - LAST_UPDATE_QUEUE > datetime.timedelta(minutes=10):
|
367 |
+
# We only redownload is last update was more than 10 minutes ago, as the queue is
|
368 |
+
# updated regularly and heavy to download
|
369 |
+
download_dataset(QUEUE_REPO, EVAL_REQUESTS_PATH)
|
370 |
+
LAST_UPDATE_QUEUE = datetime.datetime.now()
|
371 |
|
372 |
webhooks_server.launch()
|