Update app.py

app.py

@@ -24,11 +24,13 @@ footer {visibility: hidden !important;}
 # Создание интерфейса с помощью Gradio
 with gr.Blocks(css=css, theme='YTheme/Sketch') as demo:
     with gr.Row():
-        code = gr.Textbox(label="Ключ доступа", type="password")
+        with gr.Row():
+            code = gr.Textbox(label="Ключ доступа", type="password")
+        with gr.Row():
+            prompt_input = gr.Textbox(label="Описание изображения")
+        submit_btn = gr.Button("Генерация", variant='primary')
     with gr.Row():
-        prompt_input = gr.Textbox(label="Описание изображения")
-    submit_btn = gr.Button("Генерация", variant='primary')
-    image_output = gr.Image(label="Изображение")
+        image_output = gr.Image(label="Изображение")
     
     submit_btn.click(fn=generate_image, inputs=[prompt_input, code], outputs=image_output)