Update demo.py

demo.py

@@ -14,7 +14,7 @@ from firebase_admin import db, credentials
 ##################################################################
 
 
-NUMBER_OF_IMAGES_PER_ROW = 4
+NUMBER_OF_IMAGES_PER_ROW = 7
 NUMBER_OF_ROWS = 2
 
 
@@ -135,7 +135,7 @@ def next():
     new_experiment = generate_new_experiment()
 
     new_img_components = [
-        gr.Image(value=img["name"], label=f"{i}", elem_id="unsel", show_label=False, show_download_button=False, show_share_button=False, interactive=False, height=200, width=200)
+        gr.Image(value=img["name"], label=f"{i}", elem_id="unsel", show_label=False, show_download_button=False, show_share_button=False, interactive=False)
         for i, img in enumerate(new_experiment["options"])
     ]
     new_corrupted_component = gr.Image(value=new_experiment["corrupted"]["name"], label="corr", elem_id="corrupted", show_label=False, show_download_button=False, show_share_button=False, interactive=False)
@@ -146,11 +146,11 @@ def on_select(evt: gr.SelectData, experiment, *img_components):  # SelectData is
     new_selected = int(evt.target.label)
 
     new_img_components = [
-        gr.Image(value=img["name"], label=f"{i}", elem_id="unsel", show_label=False, show_download_button=False, show_share_button=False, interactive=False, height=200, width=200)
+        gr.Image(value=img["name"], label=f"{i}", elem_id="unsel", show_label=False, show_download_button=False, show_share_button=False, interactive=False)
         for i, img in enumerate(experiment["options"])
     ]
     new_img_components[new_selected] = (
-        gr.Image(value=experiment["options"][new_selected]["name"], label=f"{new_selected}", elem_id="sel", show_label=False, show_download_button=False, show_share_button=False, interactive=False, height=200, width=200)
+        gr.Image(value=experiment["options"][new_selected]["name"], label=f"{new_selected}", elem_id="sel", show_label=False, show_download_button=False, show_share_button=False, interactive=False)
     )
 
     experiment["selected_image"] = int(evt.target.label)
@@ -180,7 +180,7 @@ with gr.Blocks(title="Unsupervised Image Editing", css=css) as demo:
     for row in range(NUMBER_OF_ROWS):
         with gr.Row():
             for col in range(NUMBER_OF_IMAGES_PER_ROW):
-                img_components.append(gr.Image(label=f"{row * NUMBER_OF_IMAGES_PER_ROW + col}", elem_id="unsel", show_label=False, show_download_button=False, show_share_button=False, interactive=False, height=200, width=200))
+                img_components.append(gr.Image(label=f"{row * NUMBER_OF_IMAGES_PER_ROW + col}", elem_id="unsel", show_label=False, show_download_button=False, show_share_button=False, interactive=False))
 
     btn_skip.click(partial(save, mode="skip"), [experiment, corrupted_component, *img_components], [experiment, corrupted_component, *img_components])
     btn_submit.click(partial(save, mode="save"), [experiment, corrupted_component, *img_components], [experiment, corrupted_component, *img_components])