app.py

app.py

@@ -49,7 +49,7 @@ def make_me():
                 stop_button.click(on_stop_click, inputs=None, outputs=[gen_button, stop_button])
     
     with gr.Row():
-        output = [gr.Image(label=m, min_width=250, height=250, elem_id="custom_image") for m in default_models]
+        output = [gr.Image(label=m, min_width=250, height=250, elem_id="custom_image", show_label=True, interactive=True, show_share_button=False) for m in default_models]
         current_models = [gr.Textbox(m, visible=False) for m in default_models]
         for m, o in zip(current_models, output):
             gen_event = gen_button.click(gen_fn, [m, txt_input], o)