Update run.py

run.py

@@ -51,7 +51,7 @@ def segment(im, interval_s=2):
         logits_n = (logits[0, 0] - logits[0, 0].min()) / (logits[0, 0].max() - logits[0, 0].min())
         logits_n = cv2.resize(logits_n, (im.shape[1], im.shape[0]))
         imout[..., 1] = np.clip(imout[..., 1] + logits_n * 200, 0, 254)
-    return imout, cnt  #np.flipud(im)
+    return imout, counter.count  #np.flipud(im)
 
 # with gr.Blocks() as demo:
 #     inp = gr.Image(sources=["webcam"], streaming=True)