derek-thomas
commited on
Commit
·
8bedcfb
1
Parent(s):
aef4cb3
Whitespace updates
Browse files
app.py
CHANGED
|
@@ -1,16 +1,16 @@
|
|
| 1 |
import os
|
| 2 |
from pathlib import Path
|
| 3 |
|
| 4 |
-
from fastapi import BackgroundTasks, Response, status
|
| 5 |
import gradio as gr
|
| 6 |
-
from huggingface_hub import WebhookPayload, WebhooksServer, build_hf_headers
|
| 7 |
import requests
|
|
|
|
|
|
|
| 8 |
|
|
|
|
| 9 |
from src.my_logger import setup_logger
|
|
|
|
| 10 |
from src.utilities import load_datasets, merge_and_update_datasets
|
| 11 |
from src.visualize_logs import log_file_to_html_string
|
| 12 |
-
from src.build_nomic import build_nomic
|
| 13 |
-
from src.readme_update import update_dataset_readme
|
| 14 |
|
| 15 |
proj_dir = Path(__name__).parent
|
| 16 |
|
|
@@ -80,7 +80,6 @@ async def community(payload: WebhookPayload, task_queue: BackgroundTasks):
|
|
| 80 |
except:
|
| 81 |
return Response("No task scheduled", status_code=status.HTTP_200_OK)
|
| 82 |
|
| 83 |
-
|
| 84 |
# No need to run for README updates
|
| 85 |
try:
|
| 86 |
commit_files_url = f"""{payload.repo.url.api}/compare/{payload.updatedRefs[0]['oldSha']}..{payload.updatedRefs[0].newSha}?raw=true"""
|
|
@@ -94,6 +93,7 @@ async def community(payload: WebhookPayload, task_queue: BackgroundTasks):
|
|
| 94 |
task_queue.add_task(_process_webhook, payload=payload)
|
| 95 |
return Response("Task scheduled.", status_code=status.HTTP_202_ACCEPTED)
|
| 96 |
|
|
|
|
| 97 |
def _process_webhook(payload: WebhookPayload):
|
| 98 |
logger.info(f"Loading new dataset...")
|
| 99 |
dataset, original_dataset = load_datasets()
|
|
@@ -118,6 +118,7 @@ def _process_webhook(payload: WebhookPayload):
|
|
| 118 |
|
| 119 |
logger.info(f"Update from webhook completed!")
|
| 120 |
|
|
|
|
| 121 |
if __name__ == '__main__':
|
| 122 |
app.launch(server_name="0.0.0.0", show_error=True, server_port=7860)
|
| 123 |
# ui.queue().launch(server_name="0.0.0.0", show_error=True, server_port=7860)
|
|
|
|
| 1 |
import os
|
| 2 |
from pathlib import Path
|
| 3 |
|
|
|
|
| 4 |
import gradio as gr
|
|
|
|
| 5 |
import requests
|
| 6 |
+
from fastapi import BackgroundTasks, Response, status
|
| 7 |
+
from huggingface_hub import WebhookPayload, WebhooksServer, build_hf_headers
|
| 8 |
|
| 9 |
+
from src.build_nomic import build_nomic
|
| 10 |
from src.my_logger import setup_logger
|
| 11 |
+
from src.readme_update import update_dataset_readme
|
| 12 |
from src.utilities import load_datasets, merge_and_update_datasets
|
| 13 |
from src.visualize_logs import log_file_to_html_string
|
|
|
|
|
|
|
| 14 |
|
| 15 |
proj_dir = Path(__name__).parent
|
| 16 |
|
|
|
|
| 80 |
except:
|
| 81 |
return Response("No task scheduled", status_code=status.HTTP_200_OK)
|
| 82 |
|
|
|
|
| 83 |
# No need to run for README updates
|
| 84 |
try:
|
| 85 |
commit_files_url = f"""{payload.repo.url.api}/compare/{payload.updatedRefs[0]['oldSha']}..{payload.updatedRefs[0].newSha}?raw=true"""
|
|
|
|
| 93 |
task_queue.add_task(_process_webhook, payload=payload)
|
| 94 |
return Response("Task scheduled.", status_code=status.HTTP_202_ACCEPTED)
|
| 95 |
|
| 96 |
+
|
| 97 |
def _process_webhook(payload: WebhookPayload):
|
| 98 |
logger.info(f"Loading new dataset...")
|
| 99 |
dataset, original_dataset = load_datasets()
|
|
|
|
| 118 |
|
| 119 |
logger.info(f"Update from webhook completed!")
|
| 120 |
|
| 121 |
+
|
| 122 |
if __name__ == '__main__':
|
| 123 |
app.launch(server_name="0.0.0.0", show_error=True, server_port=7860)
|
| 124 |
# ui.queue().launch(server_name="0.0.0.0", show_error=True, server_port=7860)
|