Update app.py

app.py

@@ -1,11 +1,8 @@
-## Modified from Akhaliq Hugging Face Demo
-## https://huggingface.co/akhaliq
-
 import gradio as gr
 import os
 import cv2
 
-def inference(file, mask, model, alpha, segmentation_strength):
+def inference(file, mask, model):
     im = cv2.imread(file, cv2.IMREAD_COLOR)
     cv2.imwrite(os.path.join("input.png"), im)
 
@@ -19,56 +16,52 @@ def inference(file, mask, model, alpha, segmentation_strength):
             input = i.read()
             output = remove(
                 input, 
-                session=new_session(model), 
-                only_mask=(True if mask == "Mask only" else False),
-                alpha=alpha,
-                segmentation_strength=segmentation_strength
+                session = new_session(model), 
+                only_mask = (True if mask == "Mask only" else False)
             ) 
+            
+            
+            
             o.write(output)
     return os.path.join("output.png")
-
-title = "RemBG_Super"
-description = "Gradio demo for RemBG. erase the background of any image, To use it, simply upload your image and adjust the sliders and choose a eraser plugin from the U2net library."
+        
+title = "RemBG"
+description = "Gradio demo for RemBG. To use it, simply upload your image and wait. Read more at the link below."
 article = "<p style='text-align: center;'><a href='https://github.com/danielgatis/rembg' target='_blank'>Github Repo</a></p>"
 
+
 gr.Interface(
     inference, 
     [
         gr.inputs.Image(type="filepath", label="Input"),
-        gr.Row([
-            gr.inputs.Radio(
-                [
-                    "Default", 
-                    "Mask only"
-                ], 
-                type="value",
-                default="Default",
-                label="Choices"
-            ),
-            gr.inputs.Dropdown([
-                "u2net", 
-                "u2netp", 
-                "u2net_human_seg", 
-                "u2net_cloth_seg", 
-                "silueta",
-                "isnet-general-use",
-                "isnet-anime",
-                "sam",
+        gr.inputs.Radio(
+            [
+                "Default", 
+                "Mask only"
+            ], 
+            type="value",
+            default="Default",
+            label="Choices"
+        ),
+        gr.inputs.Dropdown([
+            "u2net", 
+            "u2netp", 
+            "u2net_human_seg", 
+            "u2net_cloth_seg", 
+            "silueta",
+            "isnet-general-use",
+            "isnet-anime",
+            "sam",
             ], 
             type="value",
             default="isnet-general-use",
             label="Models"
-            ),
-        ]),
-        gr.Row([
-            gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.5, label="Alpha (Transparency)"),
-            gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.5, label="Segmentation Strength"),
-        ]),
+        ),
     ], 
     gr.outputs.Image(type="filepath", label="Output"),
     title=title,
     description=description,
     article=article,
-    examples=[["lion.png", "Default", "u2net", 0.5, 0.5], ["girl.jpg", "Default", "u2net", 0.5, 0.5], ["anime-girl.jpg", "Default", "isnet-anime", 0.5, 0.5]],
+    examples=[["lion.png", "Default", "u2net"], ["girl.jpg", "Default", "u2net"], ["anime-girl.jpg", "Default", "isnet-anime"]],
     enable_queue=True
-).launch()
+    ).launch()