fix layout

app.py

@@ -50,9 +50,6 @@ pipe = FluxControlNetInpaintingPipeline.from_pretrained(
 pipe.transformer.to(torch.bfloat16)
 pipe.controlnet.to(torch.bfloat16)
 
-# Markdown description
-MARKDOWN = """# SnapClean | Remove Unwanted Objects from Photos 🔥Model by alimama-creative"""
-
 
 @spaces.GPU()
 def process(input_image_editor):
@@ -67,7 +64,7 @@ def process(input_image_editor):
     # Process image and mask
     image = input_image_editor['background']
     mask = input_image_editor['layers'][0]
-    size = (768, 768)
+    size = (512, 512)
     image_or = image.copy()
     image = image.convert("RGB").resize(size)
     mask = mask.convert("RGB").resize(size)
@@ -115,36 +112,11 @@ demo = gr.Interface(
     fn=process,
     inputs=image,
     outputs=[output_slider, gr.File(label="output png file")],
-    title="Magic Eraser",
-)
+    title="🫧 Snap Clean 🧽",
+    description=config.DESCRIPTION,
+    footer=config.BUY_ME_A_COFFE
 
-"""
-# Gradio interface
-with gr.Blocks() as demo:
-    gr.Markdown(MARKDOWN)
-    with gr.Row():
-        with gr.Column():
-            input_image_editor_component = gr.ImageEditor(
-                label='Image',
-                type='pil',
-                sources=["upload", "webcam"],
-                image_mode='RGB',
-                layers=False,
-                brush=gr.Brush(colors=["#FFFFFF"], color_mode="fixed")
-            )
-            submit_button_component = gr.Button(
-                value='Submit', variant='primary', scale=0
-            )
-        with gr.Column():
-            output_image_component = ImageSlider(
-                type='pil', image_mode='RGB', label='Processed photo'
-            )
+)
 
-    submit_button_component.click(
-        fn=process,
-        inputs=[input_image_editor_component],
-        outputs=[output_image_component]
-    )
-"""
 
 demo.launch(debug=False, show_error=True, share=True)

config.py

@@ -6,4 +6,23 @@ DEFAULT_CONTROLNET_CONDITIONING_SCALE = 0.9
 DEFAULT_GUIDANCE_SCALE = 3.5
 DEFAULT_SEED = 124
 DEFAULT_NUM_INFERENCE_STEPS = 24
-DEFAULT_TRUE_GUIDANCE_SCALE = 3.5
+DEFAULT_TRUE_GUIDANCE_SCALE = 3.5
+
+DESCRIPTION = """  
+<div style="max-width: 600px; margin: auto; font-family: Arial, sans-serif; color: #333;">  
+r    <h1 style="text-align: center; color: #fff;">🧽🫧Snap Clean!</h1>  
+    <p style="text-align: center; font-size: 1.1em; line-height: 1.6;">  
+Remove Unwanted Objects from Photos. 
+    </p>
+    <p style="text-align: center; font-size: 1.1em; line-height: 1.6;">      
+    </p>  
+</div>  
+"""
+
+BUY_ME_A_COFFE = """  
+<a href="https://buymeacoffee.com/thesab" target="_blank">  
+    <button style="background-color: #FFDD00; border: none; color: black; padding: 10px 20px; text-align: center; text-decoration: none; display: inline-block; font-size: 16px; margin: 4px 2px; cursor: pointer; border-radius: 10px;">  
+        ☕ Buy Me a Coffee  
+    </button>  
+</a>   
+"""