Update run.py

run.py

@@ -39,11 +39,13 @@ def segment(im, interval_s=2):
     pixel_values = image_processor(im, return_tensors="pt").pixel_values.to(device)
     outputs = model(pixel_values=pixel_values)
     logits = outputs.logits.cpu().detach().numpy() ** 2
-    imout = (logits[0, 0] - logits[0, 0].min()) / (logits[0, 0].max() - logits[0, 0].min())
+    logits_n = (logits[0, 0] - logits[0, 0].min()) / (logits[0, 0].max() - logits[0, 0].min())
+    imout = im.copy()
+    imout[..., 0] = imgout[..., 0] + logits_n / 10
     return imout  #, cnt  #np.flipud(im)
 
 with gr.Blocks() as demo:
-    inp = gr.Image(sources=["webcam"], streaming=False)
+    inp = gr.Image(sources=["webcam"], streaming=True)
     inp.stream(segment, inputs=inp, outputs=[gr.Image()]) 
 
 # demo = gr.Interface(