fffiloni commited on
Commit
cdae099
·
1 Parent(s): edc46ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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
- scribble = input_image["mask"]
 
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':