fal.ai 替代方案：哪些模型能直接换过来，代码怎么改

import os
import time
import requests

API = "https://api.omniakey.com"
HEADERS = {
    "Authorization": f"Bearer {os.environ['OMNIAKEY_API_KEY']}",
    "Content-Type": "application/json",
}

# 1. 创建任务。带上 Idempotency-Key，重试也安全。
task = requests.post(
    f"{API}/v1/media/tasks",
    headers={**HEADERS, "Idempotency-Key": "switch-from-fal-ai-002"},
    json={
        "model": "gemini-3.1-flash-image",
        "type": "image",
        "operation": "generate",
        "input": {"prompt": "a ceramic coffee cup on a wooden table, seen from above"},
        "parameters": {"aspect_ratio": "1:1", "image_size": "1K"},
    },
).json()

# 2. 轮询，直到任务结束。
while task["status"] in ("queued", "processing"):
    time.sleep(3)
    task = requests.get(f"{API}/v1/media/tasks/{task['id']}", headers=HEADERS).json()

if task["status"] != "succeeded":
    raise SystemExit(task.get("error"))

# 3. 在签名链接过期前下载图片。
open("out.jpg", "wb").write(requests.get(task["assets"][0]["url"]).content)
print(task["id"], task["assets"][0]["mime_type"], task["assets"][0]["byte_size"])