Update app.py

app.py

@@ -14,7 +14,6 @@ import numpy as np
 # from sam2_mask import create_sam2_tab
 import subprocess
 import os
-# import sam2_mask
 
 subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
 HF_TOKEN = os.environ["HF_TOKEN"]
@@ -415,6 +414,7 @@ title = """<h1 align="center">Diffusers Image Outpaint</h1>
     </p>
 </div>
 """
+sam_block = gr.load(name="spaces/LPX55/SAM2_1-Image-Predictor-Masking-Tool-CPU")
 
 with gr.Blocks(css=css, fill_height=True) as demo:
     gr.Markdown("# Diffusers Inpaint and Outpaint")
@@ -492,7 +492,10 @@ with gr.Blocks(css=css, fill_height=True) as demo:
                             label="Input Image"
                         )
                         with gr.Row():
-                            prompt_input = gr.Textbox(label="Prompt (Optional)")                                
+                            with gr.Column(scale=2):
+                                prompt_input = gr.Textbox(label="Prompt (Optional)")
+                            with gr.Column(scale=1):
+                                runout_button = gr.Button("Generate")
                         with gr.Row():
                             target_ratio = gr.Radio(
                                 label="Expected Ratio",
@@ -551,9 +554,8 @@ with gr.Blocks(css=css, fill_height=True) as demo:
                                         value=50,
                                         visible=False
                                     )
-                            with gr.Row():
-                                preview_button = gr.Button("Preview Alignment")
-                                runout_button = gr.Button("Generate")
+                                with gr.Column():
+                                    preview_button = gr.Button("Preview alignment and mask")
                     with gr.Column():
                         result_outpaint = ImageSlider(
                             interactive=False,
@@ -589,16 +591,7 @@ with gr.Blocks(css=css, fill_height=True) as demo:
         #             outputs=[image_output],
         #             show_progress=True,
         #         )
-        # with gr.TabItem("SAM2.1 Segmented Mask"):
-        #     with gr.Blocks() as sam_demo:
-        #         sam_block = gr.load(name="spaces/LPX55/SAM2_1-Image-Predictor-Masking-Tool-CPU")
-            # Define the action triggered by the upload_image_input change
-            # upload_image_input.change(
-            #     fn=update_mask,
-            #     inputs=[upload_image_input],
-            #     outputs=[image_output],
-            #     show_progress=True,
-            # )
+        
         with gr.TabItem("Misc"):
             with gr.Column():
                 clear_cache_button = gr.Button("Clear CUDA Cache")