ML library

Replicate + Hugging Face

Templates for every major Replicate category and many Hugging Face pipeline types. Use in chat, call the ML API, or chain with the data lake.

Env

REPLICATE_API_TOKEN=
HUGGINGFACE_API_KEY=   # or HF_TOKEN
ALLOW_DEMO_PIPELINE_ANON=1  # local dry-run without login

API

# List categories + tasks
GET /api/openchat/ml

# Replicate text-to-image
POST /api/openchat/ml
{ "provider": "replicate", "category": "text-to-image",
  "input": { "prompt": "OpenChat logo, minimal" } }

# Hugging Face summarization
POST /api/openchat/ml
{ "provider": "huggingface", "task": "summarization",
  "inputs": "Long text…" }

# Override model (any Hub / Replicate id)
{ "provider": "huggingface", "task": "text-generation",
  "model": "meta-llama/Meta-Llama-3-8B-Instruct",
  "inputs": "Explain inflation" }

Replicate categories (17)

CategoryDefault modelAgent slug
Text → Imageblack-forest-labs/flux-schnellreplicate-text-to-image
Image → Imageblack-forest-labs/flux-kontext-proreplicate-image-to-image
Upscale / Enhancenightmareai/real-esrganreplicate-image-upscale
Background removalcjwbw/rembgreplicate-background-removal
Face restoretencentarc/gfpganreplicate-face-restore
Text → Videominimax/video-01replicate-text-to-video
Image → Videostability-ai/stable-video-diffusionreplicate-image-to-video
Video → Videolucataco/animate-diffreplicate-video-to-video
Lip-sync / Talking headdevxpy/cog-wav2lipreplicate-lip-sync
Text → Speech (Replicate)jaaari/kokoro-82mreplicate-text-to-speech
Music generationmeta/musicgenreplicate-music
Speech → Textopenai/whisperreplicate-speech-to-text
Language / LLMmeta/meta-llama-3-8b-instructreplicate-language
Vision captionsalesforce/blipreplicate-vision-caption
Object detectionyorickvp/llava-13breplicate-vision-detect
3D generationfirtoz/trellisreplicate-3d
Embeddingsandreasjansson/clip-featuresreplicate-embeddings

Hugging Face tasks (22)

TaskDefault modelAgent slug
Text generation (LLM)mistralai/Mistral-7B-Instruct-v0.3huggingface-text-generation
Summarizationfacebook/bart-large-cnnhuggingface-summarization
TranslationHelsinki-NLP/opus-mt-en-eshuggingface-translation
Text classification / sentimentdistilbert-base-uncased-finetuned-sst-2-englishhuggingface-text-classification
Zero-shot classificationfacebook/bart-large-mnlihuggingface-zero-shot-classification
NER / token classificationdslim/bert-base-NERhuggingface-token-classification
Extractive QAdeepset/roberta-base-squad2huggingface-question-answering
Fill-maskbert-base-uncasedhuggingface-fill-mask
Sentence similaritysentence-transformers/all-MiniLM-L6-v2huggingface-sentence-similarity
Embeddings / featuressentence-transformers/all-mpnet-base-v2huggingface-feature-extraction
Text → Imageblack-forest-labs/FLUX.1-schnellhuggingface-text-to-image
Image classificationgoogle/vit-base-patch16-224huggingface-image-classification
Object detectionfacebook/detr-resnet-50huggingface-object-detection
Image segmentationfacebook/detr-resnet-50-panoptichuggingface-image-segmentation
Image captioningSalesforce/blip-image-captioning-basehuggingface-image-to-text
Visual QAdandelin/vilt-b32-finetuned-vqahuggingface-visual-question-answering
Document QAimpira/layoutlm-document-qahuggingface-document-question-answering
Speech recognition (ASR)openai/whisper-large-v3huggingface-automatic-speech-recognition
Audio classificationMIT/ast-finetuned-audioset-10-10-0.4593huggingface-audio-classification
Text → Speechespnet/kan-bayashi_ljspeech_vitshuggingface-text-to-speech
Table QAgoogle/tapas-base-finetuned-wtqhuggingface-table-question-answering
Conversationalfacebook/blenderbot-400M-distillhuggingface-conversational

Any Inference-compatible Hub model can override the default via the model field — that is how we connect “many many” models without one file per weight.

Grow further

  • Add a Replicate preset in connectors/replicate.ts → auto agent in registry
  • Add an HF task/model in connectors/huggingface.ts
  • Bundle multi-step agents in registry/ml-models.ts
Growth guide →
ML models — Replicate categories + Hugging Face tasks | OpenChat-1