Clean up

app.py

@@ -87,7 +87,6 @@ def create_advanced_demo(model: Model) -> gr.Blocks:
                              num_steps,
                              randomize_seed,
                              seed,
-                             show_denoising,
                          ],
                          outputs=[
                              result,

model.py

@@ -143,19 +143,15 @@ class Model:
         return res, out_file.name
 
     def run(self, model_name: str, scheduler_type: str, num_steps: int,
-            randomize_seed: bool, seed: int, visualize_denoising: bool
-            ) -> tuple[PIL.Image.Image, int, str | None]:
+            randomize_seed: bool,
+            seed: int) -> tuple[PIL.Image.Image, int, str | None]:
         self.set_pipeline(model_name, scheduler_type)
         if scheduler_type == 'PNDM':
             num_steps = max(4, min(num_steps, 100))
         if randomize_seed:
             seed = self.rng.randint(0, 100000)
-
-        if not visualize_denoising:
-            return self.generate(seed, num_steps)[0], seed, None
-        else:
-            res, filename = self.generate_with_video(seed, num_steps)
-            return res, seed, filename
+        res, filename = self.generate_with_video(seed, num_steps)
+        return res, seed, filename
 
     @staticmethod
     def to_grid(images: list[PIL.Image.Image],