Update

model.py

@@ -38,7 +38,7 @@ CONTROLNET_MODEL_IDS = {
 }
 
 
-def download_all_controlnet_weights():
+def download_all_controlnet_weights() -> None:
     for model_id in CONTROLNET_MODEL_IDS.values():
         ControlNetModel.from_pretrained(model_id)
 
@@ -101,7 +101,7 @@ class Model:
         num_steps: int,
         guidance_scale: float,
         seed: int,
-    ):
+    ) -> list[PIL.Image.Image]:
         generator = torch.Generator().manual_seed(seed)
         return self.pipe(prompt=prompt,
                          negative_prompt=negative_prompt,
@@ -109,7 +109,7 @@ class Model:
                          num_images_per_prompt=num_images,
                          num_inference_steps=num_steps,
                          generator=generator,
-                         image=control_image)
+                         image=control_image).images
 
     def process(
         self,
@@ -123,7 +123,7 @@ class Model:
         num_steps: int,
         guidance_scale: float,
         seed: int,
-    ):
+    ) -> list[PIL.Image.Image]:
         self.load_controlnet_weight(task_name)
         results = self.run_pipe(
             prompt=self.get_prompt(prompt, additional_prompt),
@@ -134,7 +134,7 @@ class Model:
             guidance_scale=guidance_scale,
             seed=seed,
         )
-        return [vis_control_image] + results.images
+        return [vis_control_image] + results
 
     @staticmethod
     def preprocess_canny(