Fix modifier class

This commit is contained in:
Rob Ballantyne
2025-08-20 18:06:02 +01:00
parent f9fdf04884
commit 58b078f908
+3 -3
View File
@@ -50,7 +50,7 @@ def call_text2image_workflow(
payload = { payload = {
"input": { "input": {
"request_id": str(uuid.uuid4()), "request_id": str(uuid.uuid4()),
"modifier": "RawWorkflow", # or whatever your Text2Image modifier is called "modifier": "Text2Image",
"modifications": { "modifications": {
"prompt": "a beautiful landscape with mountains and lakes", "prompt": "a beautiful landscape with mountains and lakes",
"width": 1024, "width": 1024,
@@ -73,7 +73,7 @@ def call_text2image_workflow(
verify=get_cert_file_path(), verify=get_cert_file_path(),
) )
response.raise_for_status() response.raise_for_status()
print_truncate_res(str(response.json())) print(str(response.json()))
if __name__ == "__main__": if __name__ == "__main__":
@@ -95,4 +95,4 @@ if __name__ == "__main__":
except Exception as e: except Exception as e:
log.error(f"Error during API call: {e}") log.error(f"Error during API call: {e}")
else: else:
log.error(f"Failed to get API key for endpoint {args.endpoint_group_name}") log.error(f"Failed to get API key for endpoint {args.endpoint_group_name}")