python-osam 0.5.0-1
Basename is uppercase README.mdShell command-substitution syntax
SHA-256bdba018f0d8555fef92c8ef7a4221cc47790f0517fe88d43d3e3214cead7f3d2
MaleculeH₂(Cm₃Po)Md(Bk)
Evidence
1import datetime
2import io
3import json
4import os
5import sys
6
7import click
8import imgviz
9import numpy as np
10import PIL.Image
11from loguru import logger
12from numpy.typing import NDArray
13
14from . import __version__
15from . import _humanize
16from . import _tabulate
17from . import apis
18from . import types
19
20
21@click.group(context_settings=dict(help_option_names=["-h", "--help"]))
22@click.version_option( …
22:26… ersion__)
23def cli():
24 logger.remove(0)
25 logger.add(
26 sys.stderr,
27 level="INFO",
28 colorize=True,
29 format="<level>{message}</level>",
30 backtrace=False,
31 diagnose=False,
32 )
33 os.makedirs(os.path.expanduser("~/.cache/osam"), exist_ok=True)
34 logger.add(
35 os.path.expanduser("~/.cache/osam/osam.log"), colorize=True, level="DEBUG"
⋯6 lines
52:22… bcommand_obj.get_help(ctx))
53
54
55@cli.command(help="List models")
56@click.option("--all", "-a", "show_all", is_flag=True, help="show all models")
57def list(show_all):
58 rows = []
59 for model_type in apis.registered_model_types:
60 size = model_type.get_size()
61 modified_at = model_ty …
98:58… er on file changes")
99def serve(reload):
100 try:
101 import uvicorn # type: ignore[unresolved-import]
102
103 import osam._server # noqa: F401
104
105 logger.info("Starting server...")
106 uvicorn.run("osam._server:app", host="127.0.0.1", port=11368, reload=reload)
107 except ImportError:
108 logger.error("Run `pip install osam[serve]` to use `osam serve`")
109 sys.exit(1)
110
111
112@cli.command(help="Run a model")
113@click.argument("model_name", metavar="model", type=str)
114@click.option(
115 "--image",
116 …
123:19… : str, image_path: str, prompt, json: bool) -> None:
124 image: np.ndarray = np.asarray(PIL.Image.open(image_path))
125
126 try:
127 request: types.GenerateRequest = types.GenerateRequest(
128 model=model_name, image=imag …
No evidence locations were recorded for this file. Raw result