mirror of
https://github.com/gradio-app/gradio.git
synced 2024-11-21 01:01:05 +08:00
ceb4005bb6
- Removes assumption/requirement that the script is run specifically in the demo/ directory. - Removes assumption/requirement that there are 3 unwanted large demos that should be removed. (Note: currently, the `.gradio/flagged` demo isn't there, so it will throw an error.)
39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
"""Opens X demos randomly for quick inspection
|
|
|
|
Usage: python random_demos.py <num_demos>
|
|
Example: python random_demos.py 8
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import importlib
|
|
import pathlib
|
|
import os
|
|
import random
|
|
|
|
import gradio as gr
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("num_demos", help="number of demos to launch", type=int, default=4)
|
|
args = parser.parse_args()
|
|
|
|
# get the list of directory names
|
|
demos_list = next(os.walk(pathlib.Path(__file__).parent))[1]
|
|
|
|
# Some demos are just too large or need to be run in a special way, so we'll just skip them
|
|
large_demos = ['streaming_wav2vec', 'blocks_neural_instrument_coding', '.gradio/flagged']
|
|
for large_demo in large_demos:
|
|
if large_demo in demos_list:
|
|
demos_list.remove(large_demo)
|
|
|
|
for d, demo_name in enumerate(random.sample(demos_list, args.num_demos)):
|
|
print(f"Launching demo {d+1}/{args.num_demos}: {demo_name}")
|
|
# import the run.py file from inside the directory specified by args.demo_name
|
|
run = importlib.import_module(f"{demo_name}.run")
|
|
demo: gr.Blocks = run.demo
|
|
if d == args.num_demos - 1:
|
|
demo.launch(prevent_thread_lock=False, inbrowser=True) # prevent main thread from exiting
|
|
else:
|
|
demo.launch(prevent_thread_lock=True, inbrowser=True)
|