Update app.py

app.py

@@ -211,9 +211,11 @@ with gr.Blocks(css=css) as demo:
     )
     
     with gr.Row(elem_id="prompt-container"):
-        btn_refresh = gr.Button(value="Refresh to get deployed models")
-        btn_refresh.click(None, js="window.location.reload()")
-        current_model = gr.Dropdown(label="Current Model", choices=list_models, value=list_models[0])
+        with gr.Column():
+            current_model = gr.Dropdown(label="Current Model", choices=list_models, value=list_models[0])
+        with gr.Column(scale = 0.1):
+            btn_refresh = gr.Button(value="Refresh to get deployed models")
+            btn_refresh.click(None, js="window.location.reload()")
         
     with gr.Row(elem_id="prompt-container"):
         text_prompt = gr.Textbox(label="Prompt", placeholder="a cute dog", value = "a lovely cat", lines=1, elem_id="prompt-text-input")