Update app.py

app.py

@@ -193,8 +193,7 @@ EXPLANATION = """\n\n
 1. Click on 'Start'
 2. A prompt and 4 different images are displayed
 3. Select your favorite image
-4. Click on 'Confirm'
-5. After 10 rounds your favorite diffusion model is displayed
+4. After 10 rounds your favorite diffusion model is displayed
 """
 
 GALLERY_COLUMN_NUM = len(SUBMISSIONS)
@@ -237,20 +236,20 @@ with gr.Blocks() as demo:
     with gr.Column(visible=False) as gallery_view:
         counter = gr.Markdown(f" ### 1/{NUM_QUESTIONS}")
         prompt = gr.Markdown(PROMPT_FORMAT.format(""))
-        gallery = gr.Gallery(
-            label="All images", show_label=False, elem_id="gallery"
-        ).style(columns=GALLERY_COLUMN_NUM, object_fit="contain", preview=True)
         with gr.Blocks():
             with gr.Row() as images:
-                image_1 = gr.Image(interactive=False)
-                image_2 = gr.Image(interactive=False)
-                image_3 = gr.Image(interactive=False)
-                image_4 = gr.Image(interactive=False)
-            with gr.Row():
-                image1_button = gr.Button("1").style(full_width=False)
-                image2_button = gr.Button("2").style(full_width=False)
-                image3_button = gr.Button("3").style(full_width=False)
-                image4_button = gr.Button("4").style(full_width=False)
+                with gr.Column:
+                    image_1 = gr.Image(interactive=False)
+                    image_1_button = gr.Button("Select 1").style(full_width=True)
+                with gr.Column:
+                    image_2 = gr.Image(interactive=False)
+                    image_2_button = gr.Button("Select 2").style(full_width=True)
+                with gr.Column:
+                    image_3 = gr.Image(interactive=False)
+                    image_3_button = gr.Button("Select 3").style(full_width=True)
+                with gr.Column:
+                    image_3 = gr.Image(interactive=False)
+                    image_3_button = gr.Button("Select 4").style(full_width=True)         
                 
         next_button = gr.Button("Confirm").style(full_width=True)