switch to gradio Blocks mode

app.py

@@ -79,18 +79,34 @@ def run(ref_path, ref_style, ref_prompt, prompt1, prompt2, prompt3):
 
     return images_a
 
-gr.Interface(
-    fn=run,
-    inputs=[
-        gr.Image(type="filepath", value="./example_image/medieval-bed.jpeg"),
-        gr.Textbox(value="medieval painting"),
-        gr.Textbox(value="Man laying on bed"),
-        gr.Textbox(value="A man working on a laptop"),
-        gr.Textbox(value="A man eating pizza"),
-        gr.Textbox(value="A woman playing on saxophone")
-    ],
-    outputs=[
-        gr.Gallery()
-    ],
-    title="Google's Style Aligned Transfer"
-).launch()
+with gr.Blocks() as demo:
+    with gr. Column():
+        gr.HTML("""
+        <h2 style="text-align: center;">Google's StyleAligned Transfer</h2>
+        """            
+        )
+    with gr.Row():
+        with gr.Column():
+            ref_path = gr.Image(type="filepath", value="./example_image/medieval-bed.jpeg")
+        with gr.Column():
+            ref_style = gr.Textbox(value="medieval painting")
+            ref_prompt = gr.Textbox(value="Man laying on bed")
+            prompt1 = gr.Textbox(value="A man working on a laptop")
+            prompt2 = gr.Textbox(value="A man eating pizza")
+            prompt3 = gr.Textbox(value="A woman playing on saxophone")
+            run_button = gr.Button("Submit")
+        with gr.Column():
+            results = gr.Gallery()
+
+    run_button.click(
+        fn = run,
+        inputs = [
+            ref_path, ref_style, ref_prompt,
+            prompt1, prompt2, prompt3    
+        ],
+        outputs = [
+            results
+        ]
+    )
+
+demo.queue().launch()