ComfyUI/test_api_runpod.py
2023-08-09 00:06:00 +05:30

40 lines
1.2 KiB
Python

import requests
import time
import json
# Load data from external JSON file
with open("test_input_ex.json", "r") as json_file:
data = json.load(json_file)
url = "https://api.runpod.ai/v2/vuusai5vaeia2x/runsync"
headers = {
"Content-Type": "application/json",
"Authorization": "Bearer T1530GW4R81PRD3MHSQGHJOI9D3GZG5Y4K7BTK0G"
}
response = requests.post(url, json=data, headers=headers)
response_data = response.json()
print("Response Data",response_data)
# if "run_id" in response_data:
# run_id = response_data["run_id"]
# print(f"Successfully started run with ID: {run_id}")
# # Now, let's wait for the result
# while True:
# status_response = requests.get(f"{url}/{run_id}", headers=headers)
# status_data = status_response.json()
# if status_data.get("status") == "finished":
# print("Run finished!")
# # You can access the result in status_data["result"]
# break
# elif status_data.get("status") == "failed":
# print("Run failed.")
# break
# time.sleep(5) # Wait for 5 seconds before checking again
# else:
# print("Failed to start the run.")