Update app.py

app.py

@@ -50,7 +50,7 @@ with gr.Blocks() as demo:
         gen_button.click(lambda s: gr.update(interactive = True), None, stop_button)
            
         with gr.Column():
-            output = [gr.Image(label = m, min_width=640) for m in default_models]
+            output = [gr.Image(label = m, max_width=640) 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):