space

app.py

@@ -315,7 +315,6 @@ def create_demo(model):
             seed, cfg, steps, upsampling_steps, upsampling_noise, layout_text, task_text, content_text = inputs[-8:]
                 
             results = generate(
-                model, 
                 images, 
                 [layout_text, task_text, content_text], 
                 seed=seed, cfg=cfg, steps=steps, 
@@ -462,8 +461,20 @@ def create_demo(model):
 
 
 @spaces.GPU
-def generate(model, **args):
-    return model.process_images(args)
+def generate(
+    images, 
+    prompts, 
+    seed, cfg, steps, 
+    upsampling_steps, upsampling_noise):
+    return model.process_images(
+        images=images, 
+        prompts=prompts, 
+        seed=seed, 
+        cfg=cfg, 
+        steps=steps, 
+        upsampling_steps=upsampling_steps, 
+        upsampling_noise=upsampling_noise)
+
 
 def parse_args():
     parser = argparse.ArgumentParser()