bug fix

app.py

@@ -55,17 +55,10 @@ with demo:
         gr.Markdown("""
         # <center>🖼️ Image Watermark Remover</center>
         """)
-        with gr.Row():
-            with gr.Row():
-                input_image_watermark = gr.Image(label="Upload an Image", tool="sketch", type="pil", interactive=True)
-            with gr.Row():
-                with gr.Column():
-                    pass
-                with gr.Column():
-                    image_remove_btn = gr.Button("Remove Watermark", interactive=True, elem_id="remove_btn")
-                with gr.Column():
-                    pass
+        input_image_watermark = gr.Image(label="Upload an Image", tool="sketch", type="pil", interactive=True)
+        image_remove_btn = gr.Button("Remove Watermark", interactive=True, elem_id="remove_btn")
         output_image_clean = gr.Image(label="Output Image", interactive=True)
+        
         image_remove_btn.click(remove_image_watermark, inputs=[input_image_watermark], outputs=[output_image_clean])
 
     with gr.Tab("Video Watermark Remover"):

image_watermark_remover.py

@@ -3,14 +3,14 @@ import requests
 from PIL import Image
 
 def remove_image_watermark(input):
-    image = input["image"]
-    mask = input["mask"]
+    image = input["image"].convert("RGB")
+    mask = input["mask"].convert("RGB")
     image_data = io.BytesIO()
-    image.save(image_data, format=image.format)
+    image.save(image_data, format="JPEG")
     image_data = image_data.getvalue()
 
     mask_data = io.BytesIO()
-    mask.save(mask_data, format=mask.format)
+    mask.save(mask_data, format="JPEG")
     mask_data = mask_data.getvalue()
 
     # Prepare form data
@@ -48,8 +48,8 @@ def remove_image_watermark(input):
         }
 
     files_data = {
-        'image': ('image.png', image_data),
-        'mask': ('mask.png', mask_data)
+        'image': ('image.jpg', image_data),
+        'mask': ('mask.jpg', mask_data),
     }
 
     response = requests.post('https://ahmedghani-lama-cleaner-lama.hf.space/inpaint', data=form_data, files=files_data)