change layout a bit

app.py

@@ -65,11 +65,8 @@ for detailed information about how to use and fine-tune PaliGemma and PaliGemma
 with gr.Blocks(css="style.css") as demo:
     gr.Markdown(INTRO_TEXT)
     with gr.Column():
+        image = gr.Image(label="Input Image", type="pil", height=400)
         question = gr.Text(label="Question")
-        image = gr.Image(label="Input Image", type="pil", height=500)
-        caption_btn = gr.Button(value="Submit")
-        text_output = gr.Text(label="Text Output")
-        
         tokens = gr.Slider(
             label="Max New Tokens",
             info="Set to larger for longer generation.",
@@ -78,6 +75,12 @@ with gr.Blocks(css="style.css") as demo:
             value=80,
             step=10,
         )
+        
+        
+        caption_btn = gr.Button(value="Submit")
+        text_output = gr.Text(label="Text Output")
+        
+        
     
     caption_inputs = [
         question,