Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -18,14 +18,14 @@ birefnet = AutoModelForImageSegmentation.from_pretrained(
|
|
18 |
)
|
19 |
|
20 |
|
21 |
-
RMBG2 = AutoModelForImageSegmentation.from_pretrained(
|
22 |
-
|
23 |
-
)
|
24 |
|
25 |
# Keep them in a dict to switch easily
|
26 |
models_dict = {
|
27 |
"BiRefNet": birefnet,
|
28 |
-
"RMBG-2.0": RMBG2
|
29 |
}
|
30 |
|
31 |
# Transform
|
@@ -101,21 +101,27 @@ def process_file(file_path: str, model_choice: str):
|
|
101 |
|
102 |
# GRadio UI
|
103 |
|
104 |
-
model_selector_1 = gr.Radio(
|
105 |
-
|
106 |
-
|
107 |
-
|
108 |
-
)
|
109 |
-
model_selector_2 = gr.Radio(
|
110 |
-
|
111 |
-
|
112 |
-
|
113 |
-
)
|
114 |
-
model_selector_3 = gr.Radio(
|
115 |
-
|
116 |
-
|
117 |
-
|
118 |
-
)
|
|
|
|
|
|
|
|
|
|
|
|
|
119 |
|
120 |
# Outputs for tabs 1 & 2: single processed image
|
121 |
processed_img_upload = gr.Image(label="Processed Image (Upload)", type="pil")
|
|
|
18 |
)
|
19 |
|
20 |
|
21 |
+
# RMBG2 = AutoModelForImageSegmentation.from_pretrained(
|
22 |
+
# "briaai/RMBG-2.0", trust_remote_code=True
|
23 |
+
# )
|
24 |
|
25 |
# Keep them in a dict to switch easily
|
26 |
models_dict = {
|
27 |
"BiRefNet": birefnet,
|
28 |
+
# "RMBG-2.0": RMBG2
|
29 |
}
|
30 |
|
31 |
# Transform
|
|
|
101 |
|
102 |
# GRadio UI
|
103 |
|
104 |
+
# model_selector_1 = gr.Radio(
|
105 |
+
# choices=["BiRefNet","RMBG-2.0"],
|
106 |
+
# value="BiRefNet",
|
107 |
+
# label="Select Model"
|
108 |
+
# )
|
109 |
+
# model_selector_2 = gr.Radio(
|
110 |
+
# choices=["BiRefNet","RMBG-2.0"],
|
111 |
+
# value="BiRefNet",
|
112 |
+
# label="Select Model"
|
113 |
+
# )
|
114 |
+
# model_selector_3 = gr.Radio(
|
115 |
+
# choices=["BiRefNet", "RMBG-2.0"],
|
116 |
+
# value="BiRefNet",
|
117 |
+
# label="Select Model"
|
118 |
+
# )
|
119 |
+
|
120 |
+
radio_opts = ["BiRefNet"] # single choice everywhere
|
121 |
+
|
122 |
+
model_selector_1 = gr.Radio(radio_opts, value="BiRefNet", label="Select Model")
|
123 |
+
model_selector_2 = gr.Radio(radio_opts, value="BiRefNet", label="Select Model")
|
124 |
+
model_selector_3 = gr.Radio(radio_opts, value="BiRefNet", label="Select Model")
|
125 |
|
126 |
# Outputs for tabs 1 & 2: single processed image
|
127 |
processed_img_upload = gr.Image(label="Processed Image (Upload)", type="pil")
|