Change Layout

app.py

@@ -35,18 +35,20 @@ def generate_with_mask_(image_path: str, cloth_path: str = None, prompt: str = N
 
 
 with gr.Blocks() as demo:
-    with gr.Column():
-        with gr.Row():
-            # image = gr.inputs.Image(type = "pil", label = "Input Image")
-            image_url = gr.inputs.Textbox(label = "Input Image URL")
-        with gr.Row():
-            # cloth = gr.inputs.Image(type = "pil", label = "Cloth Image")
+    gr.Markdown('# Try On Clothes Online!')
+    gr.Markdown('## Add Your Image via an Image URL or By Simply Uploading it')
+    gr.Markdown('## Paste in a link of a Product from Flipkart, Amazon or Myntra and See the Preview! You can also paste in any Image link!')
+
+    with gr.Row():
+        with gr.Column():
+            image = gr.inputs.Image(type = "pil", label = "Input Image")
+            # image_url = gr.inputs.Textbox(label = "Input Image URL")
             cloth_url = gr.inputs.Textbox(label = "Cloth Image URL")
-        with gr.Row():
-            prompt = gr.inputs.Textbox(lines=5, label="Editing Prompt")
-    with gr.Column():
+            prompt = gr.inputs.Textbox(label="Editing Prompt")
         output = gr.outputs.Image(type = "pil", label="Generated Image")
     run = gr.Button(label="Generate Preview")
-    run.click(generate_with_mask_, inputs=[image_url, cloth_url, prompt], outputs=output)
+    run.click(generate_with_mask_, inputs=[image, cloth_url, prompt], outputs=output)
+
+demo.launch()
 
 demo.launch()