Skip to main content

NeuralHive Python SDK for model access and inference

Project description

NeuralHive SDK

Python SDK client for NeuralHive model access.

Installation

pip install neuralhive

Authentication Setup

Option 1: Configure once (recommended)

neuralhive configure --api-key nh_live_your_api_key

Then use SDK without passing key in code.

Option 2: Environment variable

export NEURALHIVE_API_KEY="nh_live_your_api_key"

Option 3: Pass key directly in code

client = Client("nh_live_your_api_key")

Quick Start

from neuralhive import Client

client = Client()
models = client.list_models()
print(models)

Finetune

Start finetuning

  • Run neuralhive finetune.
  • Select a task, model, and dataset.
  • Review the final cost and start training.

Dataset requirements

  • For slm, provide a file path.
  • For other finetune tasks, provide a directory path.
  • You can save a local dataset for future use.

Track progress

  • Run neuralhive trainings.
  • View active training names and progress.
  • You can leave training running in the background.

Usage

1. Get model details

model = client.get_model("your_model_id")
print(model)

2. Run model inference

from neuralhive import Client

client = Client()

result = client.detect(
    image_path="/absolute/path/to/image.jpg",
)

For all other methods, see API Reference below.

API Reference

Client(api_key: str | None = None)

Parameter Type Required Description
api_key str No Your NeuralHive API key

Returns:

  • Client instance

Notes:

  • Authentication is validated automatically on first API call.
  • Production recommendation: avoid hardcoded API keys in source code.

client.list_models()

Parameters:

  • None

Returns:

  • list[dict] containing allowed models for this API key

client.get_model(model_id)

Parameter Type Required Description
model_id str Yes Target model identifier

Returns:

  • dict with model details

client.detect(image_path, model_id=None, prompt=None, export_to=None)

Parameter Type Required Description
image_path str Yes Local image file path
model_id str No Target model identifier
prompt str No Optional prompt (required for SAM-like IDs)
export_to str No Export directory path for box-rendered image

Notes:

  • If model_id is not provided, SDK uses default object_detection model from your allowed models.

Returns:

  • dict

client.segment(image_path, model_id=None, prompt=None, export_to=None)

Parameter Type Required Description
image_path str Yes Local image file path
model_id str No Target model identifier
prompt str No Optional prompt (required for SAM-like IDs)
export_to str No Export directory path for mask-rendered image

Returns:

  • dict

Notes:

  • If model_id is not provided, SDK uses default instance_segmentation model from your allowed models.

client.detect_and_segment(image_path, prompt, export_to=None)

Parameter Type Required Description
image_path str Yes Local image file path
prompt str Yes Segmentation prompt
export_to str No Export directory path for mask+box-rendered image

Notes:

  • Uses SAM3 / prompt-based segmentation models.

Returns:

  • dict

client.recognize(image_path)

Parameter Type Required Description
image_path str Yes Local image file path

Returns:

  • list[list[float]] (face embeddings per detected face)

client.reidentify(image_path, model_id=None)

Parameter Type Required Description
image_path str Yes Local image file path
model_id str No Target model identifier

Returns:

  • dict

client.estimate_pose(image_path, export_to=None)

Parameter Type Required Description
image_path str Yes Local image file path
export_to str No Export directory path for pose-rendered image

Returns:

  • dict with:
  • keypoints: list[object]
  • scores: list[object]
  • saved_path: str | None

client.embed(text=None, image_path=None, model_id=None)

Parameter Type Required Description
text str No Text input
image_path str No Local image file path
model_id str No Target model identifier

Notes:

  • Exactly one input is required: text or image_path
  • If model_id is not provided, SDK uses the default clip model from your allowed models.

Returns:

  • list[float] or list[list[float]]

client.generate_text(prompt, model_id=None, image_paths=None, video_paths=None)

Parameter Type Required Description
prompt str Yes Text prompt
model_id str No Target model identifier
image_paths str or list[str] No Local image file path(s)
video_paths str or list[str] No Local video path(s) or public URL(s)

Notes:

  • If model_id is not provided, SDK uses default vlm model from your allowed models.

Returns:

  • str

client.analyze(text=None, image_paths=None, video_paths=None)

Parameter Type Required Description
text str No Text input
image_paths str or list[str] No Local image file path(s)
video_paths str or list[str] No Local video path(s) or public URL(s)

Notes:

  • At least one input is required: text or image_paths or video_paths

Returns:

  • str

client.generate_image(prompt, export_to=None)

Parameter Type Required Description
prompt str Yes Text prompt
export_to str No Export directory path (filename comes from response URL)

Returns:

  • dict with:
  • download_url: str
  • saved_path: str | None

client.edit_image(image_path, prompt, export_to=None)

Parameter Type Required Description
image_path str Yes Local image file path
prompt str Yes Text instruction for the edit
export_to str No Export directory path (filename comes from response URL)

Returns:

  • dict with:
  • download_url: str
  • saved_path: str | None

client.transcribe_audio(audio_path)

Parameter Type Required Description
audio_path str Yes Local audio file path

Returns:

  • str

client.generate_3d(image_path, export_to=None)

Parameter Type Required Description
image_path str Yes Local image file path
export_to str No Export directory path (filename comes from response URL)

Returns:

  • dict with:
  • download_url: str
  • saved_path: str | None

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

neuralhive-1.1.7.tar.gz (45.9 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

neuralhive-1.1.7-py3-none-any.whl (65.7 kB view details)

Uploaded Python 3

File details

Details for the file neuralhive-1.1.7.tar.gz.

File metadata

  • Download URL: neuralhive-1.1.7.tar.gz
  • Upload date:
  • Size: 45.9 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.2

File hashes

Hashes for neuralhive-1.1.7.tar.gz
Algorithm Hash digest
SHA256 52b850d6a44ce7efbf17db01543aa3315c62ec1d33b2c88d752617beefde2b99
MD5 1ecf5a610e8e0a5aa545c45115687f1c
BLAKE2b-256 aab9cc6cdab9c4acbdb264ffce323fde672ec28608da745b615aba4a13622747

See more details on using hashes here.

File details

Details for the file neuralhive-1.1.7-py3-none-any.whl.

File metadata

  • Download URL: neuralhive-1.1.7-py3-none-any.whl
  • Upload date:
  • Size: 65.7 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.2

File hashes

Hashes for neuralhive-1.1.7-py3-none-any.whl
Algorithm Hash digest
SHA256 0619475e31c0b55fe68aee3c5553d6d6e8286a5ba74475748697405e6254040b
MD5 24fab6aec122b3fb28e800236f4a70cf
BLAKE2b-256 21bf3fbb2a0fafed8438c1c123008c14b04df625c1277e2f0cf773d3d0f145ff

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page