mindchain's picture
Upload README.md with huggingface_hub
4c0054c verified
metadata
title: NeMo DataDesigner API
emoji: 🎨
colorFrom: blue
colorTo: purple
sdk: docker
pinned: false

NVIDIA NeMo DataDesigner API

REST API for synthetic data generation using NVIDIA NeMo DataDesigner with z.ai backend.

Endpoints

Method Endpoint Description
GET / Health check
GET /health Health check
GET /models List available models
GET /sampler-types List sampler types
POST /generate Generate synthetic data
POST /preview Preview single record

Usage

Generate Data

curl -X POST https://mindchain-nemo-datadesigner-api.hf.space/generate \
  -H "Content-Type: application/json" \
  -d '{
    "num_records": 5,
    "model": "glm-4.7",
    "columns": [
      {
        "name": "category",
        "type": "sampler",
        "params": {
          "sampler_type": "CATEGORY",
          "values": ["Electronics", "Books", "Clothing"]
        }
      },
      {
        "name": "review",
        "type": "llm_text",
        "params": {
          "prompt": "Write a brief product review for a {{ category }} item."
        }
      }
    ]
  }'

Preview

curl -X POST https://mindchain-nemo-datadesigner-api.hf.space/preview \
  -H "Content-Type: application/json" \
  -d '{
    "model": "glm-4.7",
    "columns": [
      {"name": "category", "type": "sampler", "params": {"sampler_type": "CATEGORY", "values": ["A", "B"]}},
      {"name": "text", "type": "llm_text", "params": {"prompt": "Generate text about {{ category }}"}}
    ]
  }'

Available Models

ID Name Use Case
glm-5 GLM-5 (Opus) Complex reasoning
glm-4.7 GLM-4.7 (Sonnet) General purpose
glm-4.5-air GLM-4.5-Air (Haiku) Fast generation

Column Types

Type Description Required Params
sampler Statistical samplers sampler_type, type-specific params
llm_text LLM text generation prompt
llm_code Code generation prompt, language
llm_structured Structured JSON prompt, schema