Update run.py

run.py

@@ -48,7 +48,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, cnt  #np.flipud(im)
 
 # with gr.Blocks() as demo:
 #     inp = gr.Image(sources=["webcam"], streaming=True)
@@ -57,7 +57,7 @@ def segment(im, interval_s=2):
 demo = gr.Interface(
     segment,
     [gr.Image(sources=["webcam"], streaming=True)],
-    ["image"],
+    [gr.Image(), gt.Number()],
     css=".output-image, .input-image, .image-preview {height: 400px !important}"
 )