Update run.py
Browse files
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 |
-
|
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__":
|