Update run.py

run.py

@@ -42,17 +42,15 @@ 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()]) 
-
-demo = gr.Interface(
-    segment,
-    gr.Video(sources=["webcam"]),
-    gr.Video(autoplay=True)
-    # [gr.Image(sources=["webcam"], streaming=False)],
-    # ["image"],
-)
+with gr.Blocks() as demo:
+    inp = gr.Image(sources=["webcam"], streaming=False)
+    inp.stream(segment, inputs=inp, outputs=[gr.Image()]) 
+
+# demo = gr.Interface(
+#     segment,
+#     [gr.Image(sources=["webcam"], streaming=False)],
+#     ["image"],
+# )
 
 if __name__ == "__main__":