File size: 1,198 Bytes
c446951 |
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 41 42 43 44 45 46 47 |
import inference_cli.lib
import typer
from inference_cli.server import server_app
from typing_extensions import Annotated
app = typer.Typer()
app.add_typer(server_app, name="server")
@app.command()
def infer(
image: Annotated[
str, typer.Argument(help="URL or local path of image to run inference on.")
],
project_id: Annotated[
str,
typer.Option(
"--project-id", "-p", help="Roboflow project to run inference with."
),
],
model_version: Annotated[
str,
typer.Option(
"--model-version",
"-v",
help="Version of model to run inference with.",
),
],
api_key: Annotated[
str,
typer.Option("--api-key", "-a", help="Roboflow API key for your workspace."),
],
host: Annotated[
str,
typer.Option("--host", "-h", help="Host to run inference on."),
] = "http://localhost:9001",
):
typer.echo(
f"Running inference on image {image}, using model ({project_id}/{model_version}), and host ({host})"
)
inference_cli.lib.infer(image, project_id, model_version, api_key, host)
if __name__ == "__main__":
app()
|