Update app.py
Browse files
app.py
CHANGED
@@ -142,8 +142,9 @@ def run_grounded_sam(input_image, text_prompt, task_type, background_prompt, bac
|
|
142 |
# generator.to(device)
|
143 |
|
144 |
# load image
|
145 |
-
image_ori = input_image["image"]
|
146 |
-
|
|
|
147 |
original_size = image_ori.shape[:2]
|
148 |
|
149 |
if task_type == 'text':
|
|
|
142 |
# generator.to(device)
|
143 |
|
144 |
# load image
|
145 |
+
#image_ori = input_image["image"]
|
146 |
+
image_ori = input_image
|
147 |
+
#scribble = input_image["mask"]
|
148 |
original_size = image_ori.shape[:2]
|
149 |
|
150 |
if task_type == 'text':
|