Update app.py

app.py

@@ -382,19 +382,17 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
     
     with gr.Row():
         with gr.Column():
-            with gr.Tabs() as input_tabs:
-                with gr.Tab(label="", id=0) as single_image_input_tab:
-                    #image_prompt = gr.Image(label="Image Prompt", format="png", image_mode="RGBA", type="pil", height=300)
-                    with gr.Column():    
-                        image_prompt = gr.ImageMask(type="pil", image_mode="RGB", height=512, value={"background":Image.new("RGB", (512, 512), (255, 255, 255)), "layers":[Image.new("RGB", (512, 512), (255, 255, 255))], "composite":Image.new("RGB", (512, 512), (255, 255, 255))})
-                    with gr.Row():
-                        sketch_btn = gr.Button("process sketch")
-                        generate_btn = gr.Button("Generate 3D")
-                    with gr.Row():
-                        prompt = gr.Textbox(label="Prompt")
-                        style = gr.Dropdown(label="Style", choices=STYLE_NAMES, value=DEFAULT_STYLE_NAME)
+            #image_prompt = gr.Image(label="Image Prompt", format="png", image_mode="RGBA", type="pil", height=300)
+            with gr.Column():    
+                image_prompt = gr.ImageMask(type="pil", image_mode="RGB", height=512, value={"background":Image.new("RGB", (512, 512), (255, 255, 255)), "layers":[Image.new("RGB", (512, 512), (255, 255, 255))], "composite":Image.new("RGB", (512, 512), (255, 255, 255))})
+            with gr.Row():
+                sketch_btn = gr.Button("process sketch")
+                generate_btn = gr.Button("Generate 3D")
+            with gr.Row():
+                prompt = gr.Textbox(label="Prompt")
+                style = gr.Dropdown(label="Style", choices=STYLE_NAMES, value=DEFAULT_STYLE_NAME)
                             
-                with gr.Tab(label="Multiple Images", id=1, visible=False) as multiimage_input_tab:
+            with gr.Tab(label="Multiple Images", id=1, visible=False) as multiimage_input_tab:
                     multiimage_prompt = gr.Gallery(label="Image Prompt", format="png", type="pil", height=300, columns=3)
                     gr.Markdown("""
                         Input different views of the object in separate images. 
@@ -470,12 +468,12 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
     with gr.Row() as single_image_example:
         examples = gr.Examples(
             examples=[
-                f'assets/example_image/{image}'
-                for image in os.listdir("assets/example_image")
+                f'assets/example_image_sketches/{image}'
+                for image in os.listdir("assets/example_image_sketches")
             ],
             inputs=[image_prompt],
             fn=preprocess_image,
-            outputs=[image_prompt],
+            outputs=[image_prompt_processed],
             run_on_click=True,
             examples_per_page=64,
         )
@@ -493,10 +491,10 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
     demo.load(start_session)
     demo.unload(end_session)
     
-    single_image_input_tab.select(
-        lambda: tuple([False, gr.Row.update(visible=True), gr.Row.update(visible=False)]),
-        outputs=[is_multiimage, single_image_example, multiimage_example]
-    )
+    # single_image_input_tab.select(
+    #     lambda: tuple([False, gr.Row.update(visible=True), gr.Row.update(visible=False)]),
+    #     outputs=[is_multiimage, single_image_example, multiimage_example]
+    # )
     multiimage_input_tab.select(
         lambda: tuple([True, gr.Row.update(visible=False), gr.Row.update(visible=True)]),
         outputs=[is_multiimage, single_image_example, multiimage_example]