Update app.py

app.py

@@ -31,7 +31,7 @@ def extend_choices(choices):
 
 def update_imgbox(choices):
     choices_plus = extend_choices(choices)
-    return [gr.Image(None, label = m, visible = (m != 'NA'), min_width=640) for m in choices_plus]
+    return [gr.Image(None, label = m, visible = (m != 'NA')) for m in choices_plus]
 
     
 def gen_fn(model_str, prompt):
@@ -50,7 +50,7 @@ with gr.Blocks() as demo:
         gen_button.click(lambda s: gr.update(interactive = True), None, stop_button)
            
         with gr.Row():
-            output = [gr.Image(label = m) for m in default_models]
+            output = [gr.Image(label = m, min_width=256) 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):