From daade339c4ecb2063457eecce1be5ead70b4dc16 Mon Sep 17 00:00:00 2001 From: Emerson Havener Date: Sun, 27 Nov 2022 14:36:51 -0800 Subject: [PATCH] update process method to accept list. fix rule bounds. --- examples/architext/architext_test.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/examples/architext/architext_test.py b/examples/architext/architext_test.py index 6c2c27c..34a37b5 100644 --- a/examples/architext/architext_test.py +++ b/examples/architext/architext_test.py @@ -57,10 +57,11 @@ def __init__(self): self.device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") self.model = self.model.to(self.device) - def process(self, data: ArchitextBatchElement) -> ArchitextBatchElement: + def process(self, data: 'list[ArchitextBatchElement]') -> ArchitextBatchElement: + data = data[0] imgs_comb, out_dict = prompt_to_layout(self.model, self.tokenizer, self.device, data.prompt, data.creativity) data.result = { 'image': imgs_comb, 'layout': out_dict } - return data + return [data] class ArchitextFront(GradioFront): @@ -144,7 +145,7 @@ def present(self, task): choices = self.space_names_camel_case_to_title_case(space_names), value = [], visible = False) else: - rule_string = self.rules[random.randint(0, 6)]; + rule_string = self.rules[random.randint(0, 5)] prompt = gr.update(interactive = False) creativity = gr.update(interactive = False) feedback = gr.update(interactive = True, visible = True)