shemayons commited on
Commit
d83fa6c
·
verified ·
1 Parent(s): 7d6af6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -19
app.py CHANGED
@@ -18,14 +18,14 @@ birefnet = AutoModelForImageSegmentation.from_pretrained(
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,21 +101,27 @@ def process_file(file_path: str, model_choice: str):
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
  # 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")