Update app.py

app.py

@@ -221,9 +221,9 @@ with gr.Blocks(css=css) as demo:
                                        info = "default prompt: {}".format(DEFAULT_PROMPT)
                                        )
             select_button = gr.Button("Select Prompt Generate", variant='primary', elem_id="gen-button")
-        with gr.Row():
-            btn_refresh = gr.Button(value="Click Refresh to get newly Prompt candidates")
-            btn_refresh.click(None, js="window.location.reload()")
+    with gr.Row("prompt-container"):
+        btn_refresh = gr.Button(value="Click Refresh to get newly Prompt candidates")
+        btn_refresh.click(None, js="window.location.reload()")
         
     with gr.Row():
         image_output = gr.Image(type="pil", label="Output Image", elem_id="gallery")