Upload app.py with huggingface_hub

app.py

@@ -65,6 +65,19 @@ def remove_background_color(image, hex_color, tolerance):
     alpha_channel = np.where(mask, 0, 255)
     data[:, :, 3] = alpha_channel
     return data
+def color_remover_interface(image, hex_color, tolerance, _=None):
+    """
+    Gradio interface function for color removal.
+    """
+    try:
+        if image is None:
+            raise ValueError("Please upload an image")
+        hex_color = sanitize_hex_color(hex_color)
+        processed_image = remove_background_color(image, hex_color, tolerance)
+        return processed_image  # Return only the processed image
+    except Exception as e:
+        gr.Warning(str(e))
+        return image  # Return original image in case of error
 
 # Deploy using Gradio Client
 iface = gr.Interface(