Update run.py

run.py

@@ -45,12 +45,30 @@ def segment(im, interval_s=2):
 #     inp = gr.Image(sources=["webcam"], streaming=True)
 #     inp.stream(segment, inputs=inp, outputs=[gr.Image(), gr.Number()]) 
 
-demo = gr.Interface(
-    segment,
-    [gr.Image(sources=["webcam"], streaming=True)],
-    ["image"],
-)
+# demo = gr.Interface(
+#     segment,
+#     [gr.Image(sources=["webcam"], streaming=True)],
+#     ["image"],
+# )
 
-if __name__ == "__main__":
+# if __name__ == "__main__":
+
+#     demo.launch()
+
+from gradio_webrtc import WebRTC
 
+css = """.my-group {max-width: 600px !important; max-height: 600px !important;}
+         .my-column {display: flex !important; justify-content: center !important; align-items: center !important;}"""
+
+with gr.Blocks(css=css) as demo:
+    gr.HTML(
+    )
+    with gr.Column(elem_classes=["my-column"]):
+        with gr.Group(elem_classes=["my-group"]):
+            image = WebRTC(label="Stream"n)
+        image.stream(
+            fn=segment, inputs=[image], outputs=[image], time_limit=10
+        )
+
+if __name__ == "__main__":
     demo.launch()