import requests invoke_url = "https://ai.api.nvidia.com/v1/cosmos/nvidia/cosmos-predict1-7b" fetch_url_format = "https://api.nvcf.nvidia.com/v2/nvcf/pexec/status/" headers = { "Authorization": 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, "Accept": "application/json", } payload = { "inputs": [ { "name": "command", "shape": [1], "datatype": "BYTES", "data": [ "text2world --prompt=\"A first person view from the perspective from a human sized robot as it works in a chemical plant. The robot has many boxes and supplies nearby on the industrial shelves. The camera on moving forward, at a height of 1m above the floor. Photorealistic\"" ] } ], "outputs": [ { "name": "status", "datatype": "BYTES", "shape": [1] } ] } # re-use connections session = requests.Session() response = session.post(invoke_url, headers=headers, json=payload) while response.status_code == 202: request_id = response.headers.get("NVCF-REQID") fetch_url = fetch_url_format + request_id response = session.get(fetch_url, headers=headers) response = requests.post(invoke_url, headers=headers, json=payload) response.raise_for_status() with open('result.zip', 'wb') as f: f.write(response.content)