Automatic Mask Generator Serverless API

 1# pip install "segmind>=1.1.0"
 2# export SEGMIND_API_KEY="YOUR_API_KEY"
 3import segmind
 4
 5# Async (v2): submit to the queue and block until COMPLETED.
 6# run() returns the final result dict (600s deadline, 1.0s poll by default).
 7result = segmind.run(
 8    "automatic-mask-generator",
 9    prompt="Sofa",
10    image="https://segmind-sd-models.s3.amazonaws.com/display_images/automask-ip.jpg",
11    threshold=0.2,
12    stop_on_empty_mask=False,
13    invert_mask=False,
14    return_mask=True,
15    return_alpha=False,
16    grow_mask=10,
17    seed=468685,
18    base64=False,
19)
20print(result["status"])                      # COMPLETED
21print(result.get("output"))                  # model output (e.g. media URL)
22print(result["metrics"]["inference_time"])   # server compute seconds
23
24# --- Or submit + poll manually (track request_id, control the cadence) ---
25from segmind import SegmindClient, InferenceFailed, InferenceTimeout
26
27client = SegmindClient()                      # reads SEGMIND_API_KEY
28payload = {
29    "prompt": "Sofa",
30    "image": "https://segmind-sd-models.s3.amazonaws.com/display_images/automask-ip.jpg",
31    "threshold": 0.2,
32    "stop_on_empty_mask": False,
33    "invert_mask": False,
34    "return_mask": True,
35    "return_alpha": False,
36    "grow_mask": 10,
37    "seed": 468685,
38    "base64": False,
39}
40job = client.submit_async("automatic-mask-generator", **payload)
41print(job.request_id)                         # available immediately
42try:
43    result = job.wait(timeout=600, interval=1.0)
44except InferenceTimeout as e:
45    print("still running:", e.request_id)
46except InferenceFailed as e:
47    print("failed:", e.detail)