Update run.py

run.py

@@ -42,9 +42,9 @@ def segment(im, interval_s=2):
     imout = (logits[0, 0] - logits[0, 0].min()) / (logits[0, 0].max() - logits[0, 0].min())
     return imout  #, cnt  #np.flipud(im)
 
-# with gr.Blocks() as demo:
-#     inp = gr.Image(sources=["webcam"], streaming=True)
-#     inp.stream(segment, inputs=inp, outputs=[gr.Image(), gr.Number()]) 
+with gr.Blocks() as demo:
+    inp = gr.Image(sources=["webcam"], streaming=True)
+    inp.stream(segment, inputs=inp, outputs=[gr.Image(), gr.Number()], live=True) 
 
 # demo = gr.Interface(
 #     segment,
@@ -70,14 +70,14 @@ def segment(im, interval_s=2):
             
 #         image.stream(fn=segment, inputs=[image], outputs=[image])
 
-demo = gr.Interface(
-    fn=segment,
-    inputs=[gr.Image(sources=["webcam"], streaming=True)],
-    outputs=["image"],
-    title="Image Inference",
-    cache_examples=False,
-    live=True
-)
+# demo = gr.Interface(
+#     fn=segment,
+#     inputs=[gr.Image(sources=["webcam"], streaming=True)],
+#     outputs=["image"],
+#     title="Image Inference",
+#     cache_examples=False,
+#     live=True
+# )
 
 if __name__ == "__main__":
     demo.launch()