Update app.py

app.py

@@ -18,14 +18,14 @@ birefnet = AutoModelForImageSegmentation.from_pretrained(
 )
 
 
-RMBG2 = AutoModelForImageSegmentation.from_pretrained(
-    "briaai/RMBG-2.0", trust_remote_code=True
-)
+# RMBG2 = AutoModelForImageSegmentation.from_pretrained(
+#     "briaai/RMBG-2.0", trust_remote_code=True
+# )
 
 # Keep them in a dict to switch easily
 models_dict = {
     "BiRefNet": birefnet,
-    "RMBG-2.0": RMBG2
+    # "RMBG-2.0": RMBG2
 }
 
 # Transform
@@ -101,21 +101,27 @@ def process_file(file_path: str, model_choice: str):
 
 # GRadio UI
 
-model_selector_1 = gr.Radio(
-    choices=["BiRefNet","RMBG-2.0"]
-    value="BiRefNet",
-    label="Select Model"
-)
-model_selector_2 = gr.Radio(
-    choices=["BiRefNet","RMBG-2.0"],
-    value="BiRefNet",
-    label="Select Model"
-)
-model_selector_3 = gr.Radio(
-    choices=["BiRefNet", "RMBG-2.0"],
-    value="BiRefNet",
-    label="Select Model"
-)
+# model_selector_1 = gr.Radio(
+#     choices=["BiRefNet","RMBG-2.0"],
+#     value="BiRefNet",
+#     label="Select Model"
+# )
+# model_selector_2 = gr.Radio(
+#     choices=["BiRefNet","RMBG-2.0"],
+#     value="BiRefNet",
+#     label="Select Model"
+# )
+# model_selector_3 = gr.Radio(
+#     choices=["BiRefNet", "RMBG-2.0"],
+#     value="BiRefNet",
+#     label="Select Model"
+# )
+
+radio_opts = ["BiRefNet"]            # single choice everywhere
+
+model_selector_1 = gr.Radio(radio_opts, value="BiRefNet", label="Select Model")
+model_selector_2 = gr.Radio(radio_opts, value="BiRefNet", label="Select Model")
+model_selector_3 = gr.Radio(radio_opts, value="BiRefNet", label="Select Model")
 
 # Outputs for tabs 1 & 2: single processed image
 processed_img_upload = gr.Image(label="Processed Image (Upload)", type="pil")