EmanHassan26 commited on
Commit
6487748
·
verified ·
1 Parent(s): fae3d0f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -109,10 +109,10 @@ def get_florence_segmentation(image, task_prompt="<REFERRING_EXPRESSION_SEGMENTA
109
  @spaces.GPU
110
  def generate_segmt_img(image_path):
111
  init_image = Image.open(image_path).convert("RGB")
112
- image_np = np.array(init_image)
113
 
114
  conditioning_image = get_florence_segmentation(
115
- image_np, task_prompt="<REFERRING_EXPRESSION_SEGMENTATION>",
116
  text_input="the center object square")
117
  return conditioning_image
118
 
 
109
  @spaces.GPU
110
  def generate_segmt_img(image_path):
111
  init_image = Image.open(image_path).convert("RGB")
112
+ #image_np = np.array(init_image)
113
 
114
  conditioning_image = get_florence_segmentation(
115
+ init_image, task_prompt="<REFERRING_EXPRESSION_SEGMENTATION>",
116
  text_input="the center object square")
117
  return conditioning_image
118