Update app.py

app.py

@@ -177,9 +177,8 @@ with gr.Blocks(css=css) as demo:
             </div>
         """)
     with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):    
-        with gr.Column(scale=6, min_width=600):  
-            prompt = gr.Textbox(label="Prompt", placeholder="a cute cat, 8k", show_label=True, lines=1, elem_id="prompt-text-input")
-            text_button = gr.Button("Generate", variant='primary', elem_id="gen-button")
+        prompt = gr.Textbox(label="Prompt", placeholder="a cute cat, 8k", show_label=True, lines=1, elem_id="prompt-text-input")
+        text_button = gr.Button("Generate", variant='primary', elem_id="gen-button")
     with gr.Row():    
         with gr.Column(scale=1):
             image_output = gr.Image(elem_id="gallery")