Abijith commited on
Commit
c920b87
1 Parent(s): 05a002a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -97,7 +97,7 @@ def foward_pass(image_input: np.ndarray, points: List[List[int]]) -> np.ndarray:
97
 
98
  def main_func(inputs):
99
 
100
- image_input = inputs['image']
101
  classification, points = clipseg_prediction(image_input)
102
  if classification:
103
  masks = foward_pass(image_input, points)
 
97
 
98
  def main_func(inputs):
99
 
100
+ image_input = inputs
101
  classification, points = clipseg_prediction(image_input)
102
  if classification:
103
  masks = foward_pass(image_input, points)