update

model.py

@@ -20,7 +20,7 @@ class Model:
                 guidance_scale:int = 7,
                 num_images:int = 1,
                 num_steps:int = 2,
-                ) -> list[PIL.Image.Image]:
+                ):
         seed = np.random.randint(0, np.iinfo(np.int64).max)
         generator = torch.Generator().manual_seed(seed)
         return self.pipe(prompt=prompt,