changed layout

app.py

@@ -30,15 +30,15 @@ def process_image(image):
     
     return [view(result['hr_alb']), 1 - invert(result['dif_shd']), view_scale(result['pos_res'])]
 
-interface = gr.Interface(
-    fn=process_image,
-    inputs=gr.components.Image(type="numpy"),
-    outputs=[
-        gr.components.Image(type='numpy', label="Albedo"),
-        gr.components.Image(type='numpy', label="Diffuse Shading"),
-        gr.components.Image(type='numpy', label="Residual"),
-    ],
-    live=True
-)
-
-interface.launch()
+with gr.Blocks() as demo:
+    with gr.Row():
+        input_img = gr.Image(label="Input Image")
+
+    with gr.Row():
+        alb_img = gr.Image(label="Albedo")
+        shd_img = gr.Image(label="Diffuse Shading")
+        dif_img = gr.Image(label="Diffuse Image")
+
+    input_img.change(process_image, inputs=input_img, outputs=[alb_img, shd_img, dif_img])
+
+demo.launch(show_error=True)