File size: 1,290 Bytes
0bd62e5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
"""Opens X demos randomly for quick inspection
Usage: python random_demos.py <num_demos>
Example: python random_demos.py 8
Assumes:
- This is being run from the gradio/demo/ directory
"""
from __future__ import annotations
import argparse
import importlib
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('.'))[1]
# Some demos are just too large or need to be run in a special way, so we'll just skip them
demos_list.remove('streaming_wav2vec')
demos_list.remove('blocks_neural_instrument_coding')
demos_list.remove('flagged')
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)
|