noamholz commited on
Commit
62dda50
·
verified ·
1 Parent(s): 507fb4f

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +2 -3
run.py CHANGED
@@ -47,12 +47,11 @@ def flip_periodically(im, interval_s=2):
47
  outputs = model(pixel_values=pixel_values)
48
  logits = outputs.logits.cpu().detach().numpy() ** 2
49
  counter.imout = (logits[0, 0] - logits[0, 0].min()) / (logits[0, 0].max() - logits[0, 0].min())
50
- return counter.imout #np.flipud(im)
51
 
52
  with gr.Blocks() as demo:
53
  inp = gr.Image(sources=["webcam"], streaming=True)
54
- out = gr.Image()
55
- inp.stream(flip_periodically, inputs=inp, outputs=out)
56
 
57
 
58
  if __name__ == "__main__":
 
47
  outputs = model(pixel_values=pixel_values)
48
  logits = outputs.logits.cpu().detach().numpy() ** 2
49
  counter.imout = (logits[0, 0] - logits[0, 0].min()) / (logits[0, 0].max() - logits[0, 0].min())
50
+ return counter.imout, counter.n #np.flipud(im)
51
 
52
  with gr.Blocks() as demo:
53
  inp = gr.Image(sources=["webcam"], streaming=True)
54
+ inp.stream(flip_periodically, inputs=inp, outputs=[gr.Image(), gr.Number()])
 
55
 
56
 
57
  if __name__ == "__main__":