Update

app.py

@@ -16,8 +16,15 @@ FOOTER = '<img id="visitor-badge" src="https://visitor-badge.glitch.me/badge?pag
 def create_simple_demo(model: Model) -> gr.Blocks:
     with gr.Blocks() as demo:
         run_button = gr.Button('Generate')
-        result = gr.Image(show_label=False, elem_id='result-grid')
-        run_button.click(fn=model.run_simple, inputs=None, outputs=result)
+        with gr.Tabs():
+            with gr.TabItem('Result (Superresolved)'):
+                result = gr.Image(show_label=False, elem_id='result-grid')
+            with gr.TabItem('Result (Raw)'):
+                result_raw = gr.Image(show_label=False,
+                                      elem_id='result-grid-raw')
+        run_button.click(fn=model.run_simple,
+                         inputs=None,
+                         outputs=[result, result_raw])
     return demo
 
 

model.py

@@ -180,9 +180,9 @@ class Model:
             0, 2, 1, 3, 4).reshape(nrows * h, ncols * w, 3)
         return PIL.Image.fromarray(grid)
 
-    def run_simple(self) -> PIL.Image.Image:
+    def run_simple(self) -> tuple[PIL.Image.Image, PIL.Image.Image]:
         self.set_pipeline(self.MODEL_NAMES[0], 'DDIM')
         seed = self.rng.randint(0, 1000000)
         images = self.generate(seed, num_steps=10, num_images=4)
-        images = [self.superresolve(image) for image in images]
-        return self.to_grid(images, 2)
+        superresolved = [self.superresolve(image) for image in images]
+        return self.to_grid(superresolved, 2), self.to_grid(images, 2)

style.css

@@ -5,6 +5,10 @@ div#result-grid {
   max-width: 600px;
   max-height: 600px;
 }
+div#result-grid-raw {
+  max-width: 600px;
+  max-height: 600px;
+}
 div#result {
   max-width: 400px;
   max-height: 400px;