Update

app.py

@@ -67,13 +67,14 @@ def create_advanced_demo(model: Model) -> gr.Blocks:
                                      step=1,
                                      value=1234,
                                      label='Seed')
-                    superresolve = gr.Checkbox(value=False,
-                                               label='Superresolve')
                     run_button = gr.Button('Run')
             with gr.Column():
                 with gr.Tabs():
-                    with gr.TabItem('Result'):
+                    with gr.TabItem('Result (Superresolved)'):
                         result = gr.Image(show_label=False, elem_id='result')
+                    with gr.TabItem('Result (Raw)'):
+                        result_raw = gr.Image(show_label=False,
+                                              elem_id='result-raw')
                     with gr.TabItem('Denoising Process'):
                         result_video = gr.Video(show_label=False,
                                                 elem_id='result-video')
@@ -89,10 +90,10 @@ def create_advanced_demo(model: Model) -> gr.Blocks:
                              num_steps,
                              randomize_seed,
                              seed,
-                             superresolve,
                          ],
                          outputs=[
                              result,
+                             result_raw,
                              seed,
                              result_video,
                          ])

model.py

@@ -156,17 +156,16 @@ class Model:
         return PIL.Image.open(out_file)
 
     def run(self, model_name: str, scheduler_type: str, num_steps: int,
-            randomize_seed: bool, seed: int,
-            superresolve: bool) -> tuple[PIL.Image.Image, int, str]:
+            randomize_seed: bool,
+            seed: int) -> tuple[PIL.Image.Image, PIL.Image.Image, int, str]:
         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)
         res, filename = self.generate_with_video(seed, num_steps)
-        if superresolve:
-            res = self.superresolve(res)
-        return res, seed, filename
+        superresolved = self.superresolve(res)
+        return superresolved, res, seed, filename
 
     @staticmethod
     def to_grid(images: list[PIL.Image.Image],

style.css

@@ -13,6 +13,10 @@ div#result {
   max-width: 400px;
   max-height: 400px;
 }
+div#result-raw {
+  max-width: 400px;
+  max-height: 400px;
+}
 div#result-video {
   max-width: 400px;
   max-height: 400px;