wehub-resource-sync
adf0d17497
publish / version_or_publish (push) Has been cancelled
storybook-build / changes (push) Has been cancelled
storybook-build / :storybook-build (push) Has been cancelled
Sync Gradio Skills to Hugging Face / sync-skills (push) Has been cancelled
functional / changes (push) Has been cancelled
functional / build-frontend (push) Has been cancelled
functional / functional-test-SSR=false (push) Has been cancelled
functional / functional-reload (push) Has been cancelled
js / changes (push) Has been cancelled
js / js-test (push) Has been cancelled
docs-build / changes (push) Has been cancelled
docs-build / docs-build (push) Has been cancelled
docs-build / website-build (push) Has been cancelled
functional / functional-test-SSR=true (push) Has been cancelled
hygiene / hygiene-test (push) Has been cancelled
python / changes (push) Has been cancelled
python / build (push) Has been cancelled
python / test-ubuntu-latest-flaky (push) Has been cancelled
python / test-ubuntu-latest-not-flaky (push) Has been cancelled
python / test-windows-latest-flaky (push) Has been cancelled
python / test-windows-latest-not-flaky (push) Has been cancelled
97 lines
3.1 KiB
Python
97 lines
3.1 KiB
Python
import gradio as gr
|
|
import random
|
|
import time
|
|
import tqdm
|
|
from datasets import load_dataset # type: ignore
|
|
import shutil
|
|
from uuid import uuid4
|
|
|
|
with gr.Blocks() as demo:
|
|
with gr.Row():
|
|
text = gr.Textbox()
|
|
textb = gr.Textbox()
|
|
with gr.Row():
|
|
load_set_btn = gr.Button("Load Set")
|
|
load_nested_set_btn = gr.Button("Load Nested Set")
|
|
load_random_btn = gr.Button("Load Random")
|
|
clean_imgs_btn = gr.Button("Clean Images")
|
|
wait_btn = gr.Button("Wait")
|
|
do_all_btn = gr.Button("Do All")
|
|
track_tqdm_btn = gr.Button("Bind TQDM")
|
|
bind_internal_tqdm_btn = gr.Button("Bind Internal TQDM")
|
|
|
|
text2 = gr.Textbox()
|
|
|
|
# track list
|
|
def load_set(text, text2, progress=gr.Progress()):
|
|
imgs = [None] * 24
|
|
for img in progress.tqdm(imgs, desc="Loading from list"):
|
|
time.sleep(0.1)
|
|
return "done"
|
|
load_set_btn.click(load_set, [text, textb], text2)
|
|
|
|
# track nested list
|
|
def load_nested_set(text, text2, progress=gr.Progress()):
|
|
imgs = [[None] * 8] * 3
|
|
for img_set in progress.tqdm(imgs, desc="Nested list"):
|
|
time.sleep(2)
|
|
for img in progress.tqdm(img_set, desc="inner list"):
|
|
time.sleep(0.1)
|
|
return "done"
|
|
load_nested_set_btn.click(load_nested_set, [text, textb], text2)
|
|
|
|
# track iterable of unknown length
|
|
def load_random(data, progress=gr.Progress()):
|
|
def yielder():
|
|
for i in range(0, random.randint(15, 20)):
|
|
time.sleep(0.1)
|
|
yield None
|
|
for img in progress.tqdm(yielder()):
|
|
pass
|
|
return "done"
|
|
load_random_btn.click(load_random, {text, textb}, text2)
|
|
|
|
# manual progress
|
|
def clean_imgs(text, progress=gr.Progress()):
|
|
progress(0.2, desc="Collecting Images")
|
|
time.sleep(1)
|
|
progress(0.5, desc="Cleaning Images")
|
|
time.sleep(1.5)
|
|
progress(0.8, desc="Sending Images")
|
|
time.sleep(1.5)
|
|
return "done"
|
|
clean_imgs_btn.click(clean_imgs, text, text2)
|
|
|
|
# no progress
|
|
def wait(text):
|
|
time.sleep(4)
|
|
return "done"
|
|
wait_btn.click(wait, text, text2)
|
|
|
|
# multiple progressions
|
|
def do_all(data, progress=gr.Progress()):
|
|
load_set(data[text], data[textb], progress)
|
|
load_random(data, progress)
|
|
clean_imgs(data[text], progress)
|
|
progress(None)
|
|
wait(text)
|
|
return "done"
|
|
do_all_btn.click(do_all, {text, textb}, text2)
|
|
|
|
def track_tqdm(data, progress=gr.Progress(track_tqdm=True)):
|
|
for i in tqdm.tqdm(range(5), desc="outer"):
|
|
for j in tqdm.tqdm(range(4), desc="inner"):
|
|
time.sleep(1)
|
|
return "done"
|
|
track_tqdm_btn.click(track_tqdm, {text, textb}, text2)
|
|
|
|
def bind_internal_tqdm(data, progress=gr.Progress(track_tqdm=True)):
|
|
outdir = "__tmp/" + str(uuid4())
|
|
load_dataset("beans", split="train", cache_dir=outdir)
|
|
shutil.rmtree(outdir)
|
|
return "done"
|
|
bind_internal_tqdm_btn.click(bind_internal_tqdm, {text, textb}, text2)
|
|
|
|
if __name__ == "__main__":
|
|
demo.launch()
|