Spaces:
Running
Running
File size: 1,211 Bytes
cec5823 |
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 |
from rstor.analyzis.interactive.model_selection import MODELS_PATH
import argparse
def get_models_parser(parser: argparse.ArgumentParser = None, help: str = "Inference",
default_models_path: str = MODELS_PATH) -> argparse.ArgumentParser:
if parser is None:
parser = argparse.ArgumentParser(description=help)
parser.add_argument("-e", "--experiments", type=int, nargs="+", required=True,
help="Experience indexes to be used at inference time")
parser.add_argument("-m", "--models-storage", type=str, help="Model storage path", default=default_models_path)
return parser
def get_parser(
parser: argparse.ArgumentParser = None,
help: str = "Live inference pipeline"
) -> argparse.ArgumentParser:
"""Generic parser for live interactive inference
"""
if parser is None:
parser = argparse.ArgumentParser(description=help)
get_models_parser(parser=parser, help=help)
parser.add_argument("-k", "--keyboard", action="store_true", help="Keyboard control - less sliders")
parser.add_argument("-b", "--backend", default="gradio", help="Backend to use for the GUI", choices=["gradio", "qt"])
return parser
|