Update app.py

app.py

@@ -21,6 +21,7 @@ def run(img):
     cloth_seg = generate_mask(img, net=net, palette=palette, device=device)
     return cloth_seg
 
+# CSS styling
 css = '''
 .container {max-width: 1150px;margin: auto;padding-top: 1.5rem}
 #image_upload{min-height:400px}
@@ -33,6 +34,8 @@ css = '''
 #image_upload .touch-none{display: flex}
 '''
 
+# Collect example images
+example = {}
 image_dir = 'input'
 image_list = [os.path.join(image_dir, file) for file in os.listdir(image_dir)]
 image_list.sort()
@@ -40,48 +43,23 @@ image_list.sort()
 with gr.Blocks(css=css) as demo:
     gr.HTML(read_content("header.html"))
 
-    input_method = gr.Radio(
-        choices=["Upload Image", "Use Webcam"],
-        label="Choose Input Method",
-        value="Upload Image"
-    )
-
-    upload_image = gr.Image(label="Upload Image", type="pil", visible=True)
-    webcam_input = gr.Camera(label="Webcam Image", visible=False)
+    with gr.Row():
+        with gr.Column():
+            image = gr.Image(elem_id="image_upload", type="pil", label="Input Image")
 
-    image_out = gr.Image(label="Output", elem_id="output-img")
-    btn = gr.Button("Run!")
+        with gr.Column():
+            image_out = gr.Image(label="Output", elem_id="output-img")
 
-    # Examples only apply to uploaded images
-    gr.Examples(
-        examples=image_list,
-        inputs=[upload_image],
-        label="Examples - Input Images",
-        examples_per_page=12
-    )
-
-    # Logic to toggle input components based on radio selection
-    def toggle_inputs(choice):
-        return (
-            gr.update(visible=choice == "Upload Image"),
-            gr.update(visible=choice == "Use Webcam")
+    with gr.Row():
+        gr.Examples(
+            examples=image_list,
+            inputs=[image],
+            label="Examples - Input Images",
+            examples_per_page=12
         )
+        btn = gr.Button("Run!")
 
-    input_method.change(
-        fn=toggle_inputs,
-        inputs=input_method,
-        outputs=[upload_image, webcam_input]
-    )
-
-    # Button logic: use the active image component
-    def conditional_run(uploaded, webcam, method):
-        return run(uploaded if method == "Upload Image" else webcam)
-
-    btn.click(
-        fn=conditional_run,
-        inputs=[upload_image, webcam_input, input_method],
-        outputs=[image_out]
-    )
+    btn.click(fn=run, inputs=[image], outputs=[image_out])
 
     gr.HTML(
         """
@@ -95,3 +73,5 @@ with gr.Blocks(css=css) as demo:
         </div>
         """
     )
+
+demo.launch()