Upload app.py

app.py

@@ -578,21 +578,22 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
             with gr.Row(): #.style(equal_height=True):
                 idx_chest = gr.Number(value=0, visible=False)
                 with gr.Column(scale=1, min_width=600):
-                    x_chest = gr.Image(label="Observation", interactive=False, height=HEIGHT) #).style(
+                    x_chest = gr.Image(label="Observation", interactive=False,
+                                       height=HEIGHT, width=WIDTH) #).style(
                         # height=HEIGHT
                     # )
                 with gr.Column(scale=1, min_width=600):
                     cf_x_chest = gr.Image(
                         label="Counterfactual", interactive=False
-                    , height=HEIGHT) #).style(height=HEIGHT)
+                    , height=HEIGHT, width=WIDTH) #).style(height=HEIGHT)
                 with gr.Column(scale=1, min_width=600):
                     cf_x_std_chest = gr.Image(
                         label="Counterfactual Uncertainty", interactive=False
-                    , height=HEIGHT) #).style(height=HEIGHT)
+                    , height=HEIGHT, width=WIDTH) #).style(height=HEIGHT)
                 with gr.Column(scale=1, min_width=600):
                     effect_chest = gr.Image(
                         label="Direct Causal Effect", interactive=False
-                    , height=HEIGHT) #).style(height=HEIGHT)
+                    , height=HEIGHT, width=WIDTH) #).style(height=HEIGHT)
 
             with gr.Row():
                 with gr.Column(scale=2):#.55):