noamholz commited on
Commit
8af8ad8
·
verified ·
1 Parent(s): 828c539

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +1 -1
run.py CHANGED
@@ -41,7 +41,7 @@ def segment(im, interval_s=2):
41
  logits = outputs.logits.cpu().detach().numpy() ** 2
42
  logits_n = (logits[0, 0] - logits[0, 0].min()) / (logits[0, 0].max() - logits[0, 0].min())
43
  imout = im.copy()
44
- imout[..., 0] = imgout[..., 0] + logits_n / 10
45
  return imout #, cnt #np.flipud(im)
46
 
47
  with gr.Blocks() as demo:
 
41
  logits = outputs.logits.cpu().detach().numpy() ** 2
42
  logits_n = (logits[0, 0] - logits[0, 0].min()) / (logits[0, 0].max() - logits[0, 0].min())
43
  imout = im.copy()
44
+ imout[..., 0] = imout[..., 0] + logits_n / 10
45
  return imout #, cnt #np.flipud(im)
46
 
47
  with gr.Blocks() as demo: