Update model.py

model.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import gc
 
 import numpy as np
-import PIL.Image
+from PIL import Image
 import torch
 
 from diffusers import (
@@ -107,12 +107,12 @@ class Model:
         self,
         prompt: str,
         negative_prompt: str,
-        control_image: PIL.Image.Image,
+        control_image: Image.Image,
         num_images: int,
         num_steps: int,
         controlnet_conditioning_scale: float,
         seed: int,
-    ) -> list[PIL.Image.Image]:
+    ) -> list[Image.Image]:
         generator = torch.Generator().manual_seed(seed)
         return self.pipe(
             prompt=prompt,
@@ -160,7 +160,7 @@ class Model:
         num_steps: int,
         controlnet_conditioning_scale: float,
         seed: int,
-    ) -> list[PIL.Image.Image]:
+    ) -> list[Image.Image]:
 
         # resize input_image to 1024x1024
         input_image = self.resize_image(image)