From deb64d8da19d200709b692cf89cab155b996f367 Mon Sep 17 00:00:00 2001 From: Toran Bruce Richards Date: Sun, 29 Sep 2024 20:57:05 +0100 Subject: [PATCH] run format --- .../backend/backend/blocks/replicate_flux_basic.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/autogpt_platform/backend/backend/blocks/replicate_flux_basic.py b/autogpt_platform/backend/backend/blocks/replicate_flux_basic.py index 6baca0a782..e78adb7db1 100644 --- a/autogpt_platform/backend/backend/blocks/replicate_flux_basic.py +++ b/autogpt_platform/backend/backend/blocks/replicate_flux_basic.py @@ -1,7 +1,9 @@ import replicate + from backend.data.block import Block, BlockCategory, BlockOutput, BlockSchema from backend.data.model import BlockSecret, SchemaField, SecretField + class ReplicateFluxBasicModelBlock(Block): class Input(BlockSchema): api_key: BlockSecret = SecretField( @@ -67,7 +69,7 @@ class ReplicateFluxBasicModelBlock(Block): # Run the model output = client.run( f"{model_name}", - input={"prompt": prompt,"output_format": "png"}, + input={"prompt": prompt, "output_format": "png"}, ) # Check if output is a list or a string and extract accordingly; otherwise, assign a default message @@ -76,5 +78,7 @@ class ReplicateFluxBasicModelBlock(Block): elif isinstance(output, str): result_url = output # If output is a string, use it directly else: - result_url = "No output received" # Fallback message if output is not as expected + result_url = ( + "No output received" # Fallback message if output is not as expected + ) return result_url