canceled general option to save space

app.py

@@ -27,17 +27,17 @@ Paint by Inpaint: Learning to Add Image Objects by Removing Them First</a>
 description = """
 <p style="text-align: center;">
 Gradio demo for <strong>Paint by Inpaint: Learning to Add Image Objects by Removing Them First</strong>, visit our <a href='https://rotsteinnoam.github.io/Paint-by-Inpaint/' target='_blank'>project page</a>. <br>
-The demo is both for models trained for image object addition using the <a href='https://huggingface.co/datasets/paint-by-inpaint/PIPE' target='_blank'>PIPE dataset</a> along with models trained with other datasets that are meant for general editing. <br>
+The demo involves two models: one trained for image object addition using the <a href='https://huggingface.co/datasets/paint-by-inpaint/PIPE' target='_blank'>PIPE dataset</a>, and another model further fine-tuned on the MagicBrush dataset.
 </p>
 """
 
 #   Base models
 object_addition_base_model_id = "paint-by-inpaint/add-base"
-general_editing_base_model_id = "paint-by-inpaint/general-base"
+# general_editing_base_model_id = "paint-by-inpaint/general-base"
 
 #   MagicBrush finetuned models
 object_addition_finetuned_model_id = "paint-by-inpaint/add-finetuned-mb"
-general_editing_finetuned_model_id = "paint-by-inpaint/general-finetuned-mb"
+# general_editing_finetuned_model_id = "paint-by-inpaint/general-finetuned-mb"
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.float16 if "cuda" in device else torch.float32
@@ -49,8 +49,8 @@ def load_model(model_id):
 pipe_object_addition_base = load_model(object_addition_base_model_id)
 pipe_object_addition_finetuned = load_model(object_addition_finetuned_model_id)
 
-pipe_general_editing_base = load_model(general_editing_base_model_id)
-pipe_general_editing_finetuned = load_model(general_editing_finetuned_model_id)
+# pipe_general_editing_base = load_model(general_editing_base_model_id)
+# pipe_general_editing_finetuned = load_model(general_editing_finetuned_model_id)
 
 @spaces.GPU(duration=15)
 def generate(
@@ -99,8 +99,9 @@ with gr.Blocks(css=".compact-box .gr-row { margin-bottom: 5px; } .compact-box .g
     </div>
     """.format(description=description))
 
-    with gr.Tabs():
-        with gr.Tab("Object Addition"):
+    # with gr.Tabs():
+    #     with gr.Tab("Object Addition"):
+    if 1:
             with gr.Row():
                 with gr.Column():
                     input_image = gr.Image(label="Input Image", type="pil", interactive=True)
@@ -116,7 +117,8 @@ with gr.Blocks(css=".compact-box .gr-row { margin-bottom: 5px; } .compact-box .g
 
                     with gr.Group(elem_id="compact-box"):
                         with gr.Row():
-                            steps = gr.Number(value=50, precision=0, label="Steps", interactive=True)
+                            with gr.Column():
+                                steps = gr.Number(value=50, precision=0, label="Steps", interactive=True)
                             
                             with gr.Column():
                                 with gr.Row():
@@ -130,6 +132,7 @@ with gr.Blocks(css=".compact-box .gr-row { margin-bottom: 5px; } .compact-box .g
                                     )
                         
                         with gr.Row():
+                            
                             text_cfg_scale = gr.Number(value=7.5, label="Text CFG", interactive=True)
                             image_cfg_scale = gr.Number(value=1.5, label="Image CFG", interactive=True)
 
@@ -160,65 +163,65 @@ with gr.Blocks(css=".compact-box .gr-row { margin-bottom: 5px; } .compact-box .g
                 outputs=[steps, randomize_seed, seed, text_cfg_scale, image_cfg_scale, edited_image],
             )
 
-        with gr.Tab("General Editing"):
-            with gr.Row():
-                with gr.Column():
-                    input_image_editing = gr.Image(label="Input Image", type="pil", interactive=True)
-                    instruction_editing = gr.Textbox(lines=1, label="Editing Instruction", interactive=True, max_lines=1, placeholder="Enter editing instruction here")
-
-                    model_choice_editing = gr.Radio(
-                        ["Base-Model", "Finetuned-MB-Model"],
-                        value="Base-Model",
-                        type="index",
-                        label="Choose Model",
-                        interactive=True,
-                    )
-
-                    with gr.Group(elem_id="compact-box"):
-                        with gr.Row():
-                            steps_editing = gr.Number(value=50, precision=0, label="Steps", interactive=True)
+        # with gr.Tab("General Editing"):
+        #     with gr.Row():
+        #         with gr.Column():
+        #             input_image_editing = gr.Image(label="Input Image", type="pil", interactive=True)
+        #             instruction_editing = gr.Textbox(lines=1, label="Editing Instruction", interactive=True, max_lines=1, placeholder="Enter editing instruction here")
+
+        #             model_choice_editing = gr.Radio(
+        #                 ["Base-Model", "Finetuned-MB-Model"],
+        #                 value="Base-Model",
+        #                 type="index",
+        #                 label="Choose Model",
+        #                 interactive=True,
+        #             )
+
+        #             with gr.Group(elem_id="compact-box"):
+        #                 with gr.Row():
+        #                     steps_editing = gr.Number(value=50, precision=0, label="Steps", interactive=True)
                             
-                            with gr.Column():
-                                with gr.Row():
-                                    seed_editing = gr.Number(value=2024, precision=0, label="Seed", interactive=True)
-                                    randomize_seed_editing = gr.Radio(
-                                        ["Fix Seed", "Randomize Seed"],
-                                        value="Randomize Seed",
-                                        type="index",
-                                        show_label=False,
-                                        interactive=True,
-                                    )
+        #                     with gr.Column():
+        #                         with gr.Row():
+        #                             seed_editing = gr.Number(value=2024, precision=0, label="Seed", interactive=True)
+        #                             randomize_seed_editing = gr.Radio(
+        #                                 ["Fix Seed", "Randomize Seed"],
+        #                                 value="Randomize Seed",
+        #                                 type="index",
+        #                                 show_label=False,
+        #                                 interactive=True,
+        #                             )
                         
-                        with gr.Row():
-                            text_cfg_scale_editing = gr.Number(value=7.5, label="Text CFG", interactive=True)
-                            image_cfg_scale_editing = gr.Number(value=1.5, label="Image CFG", interactive=True)
-
-                    with gr.Row():
-                        generate_button_editing = gr.Button("Generate")
-                        reset_button_editing = gr.Button("Reset")
-
-                with gr.Column():
-                    edited_image_editing = gr.Image(label="Edited Image", type="pil", interactive=False)
-
-            generate_button_editing.click(
-                fn=lambda *args: generate(*args, task_type="general_editing"),
-                inputs=[
-                    input_image_editing,
-                    instruction_editing,
-                    model_choice_editing,
-                    steps_editing,
-                    randomize_seed_editing,
-                    seed_editing,
-                    text_cfg_scale_editing,
-                    image_cfg_scale_editing,
-                ],
-                outputs=[seed_editing, text_cfg_scale_editing, image_cfg_scale_editing, edited_image_editing],
-            )
-            reset_button_editing.click(
-                fn=reset,
-                inputs=[],
-                outputs=[steps_editing, randomize_seed_editing, seed_editing, text_cfg_scale_editing, image_cfg_scale_editing, edited_image_editing],
-            )
+        #                 with gr.Row():
+        #                     text_cfg_scale_editing = gr.Number(value=7.5, label="Text CFG", interactive=True)
+        #                     image_cfg_scale_editing = gr.Number(value=1.5, label="Image CFG", interactive=True)
+
+        #             with gr.Row():
+        #                 generate_button_editing = gr.Button("Generate")
+        #                 reset_button_editing = gr.Button("Reset")
+
+        #         with gr.Column():
+        #             edited_image_editing = gr.Image(label="Edited Image", type="pil", interactive=False)
+
+        #     generate_button_editing.click(
+        #         fn=lambda *args: generate(*args, task_type="general_editing"),
+        #         inputs=[
+        #             input_image_editing,
+        #             instruction_editing,
+        #             model_choice_editing,
+        #             steps_editing,
+        #             randomize_seed_editing,
+        #             seed_editing,
+        #             text_cfg_scale_editing,
+        #             image_cfg_scale_editing,
+        #         ],
+        #         outputs=[seed_editing, text_cfg_scale_editing, image_cfg_scale_editing, edited_image_editing],
+        #     )
+        #     reset_button_editing.click(
+        #         fn=reset,
+        #         inputs=[],
+        #         outputs=[steps_editing, randomize_seed_editing, seed_editing, text_cfg_scale_editing, image_cfg_scale_editing, edited_image_editing],
+        #     )
 
     gr.Markdown(help_text)