KurtLin commited on
Commit
6cb8ce9
·
1 Parent(s): aadd02f

Initial Commit

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -36,11 +36,12 @@ def inference(image, input_label, model_choice):
36
  multimask_output=True,
37
  )
38
  mask = masks[0]
 
39
  score = scores[0]
40
  image2 = image.copy()
41
  image2[mask, 0] = 255
42
  image2[y-10:y+10, x-10:x+10, 2] = 255
43
- return score, image2
44
 
45
 
46
  my_app = gr.Blocks()
 
36
  multimask_output=True,
37
  )
38
  mask = masks[0]
39
+ print(scores)
40
  score = scores[0]
41
  image2 = image.copy()
42
  image2[mask, 0] = 255
43
  image2[y-10:y+10, x-10:x+10, 2] = 255
44
+ return f"{score}", image2
45
 
46
 
47
  my_app = gr.Blocks()