Update app.py

app.py

@@ -39,18 +39,22 @@ def segment_webcam():
     process_video(model, cap)
 
 
-def process_inputs(segment_video_checkbox, segment_webcam_checkbox, uploaded_video):
-    if segment_video_checkbox and uploaded_video:
-        segment_video(uploaded_video)
-    elif segment_webcam_checkbox:
-        segment_webcam()
+def process_inputs(start_button, stop_button, mode_selection, uploaded_video):
+    if start_button:
+        if mode_selection == "Video" and uploaded_video:
+            segment_video(uploaded_video)
+        elif mode_selection == "Webcam":
+            segment_webcam()
 
 
 iface = gr.Interface(
     fn=process_inputs,
-    inputs=[gr.inputs.Checkbox(label="Segment Video"),
-            gr.inputs.Checkbox(label="Segment Webcam"),
-            gr.inputs.File(label="Upload Video")],
+    inputs=[
+        gr.components.Button(label="Start"),
+        gr.components.Button(label="Stop"),
+        gr.components.Radio(["Video", "Webcam"], label="Mode Selection"),
+        gr.components.File(label="Upload Video")
+    ],
     outputs=None,
     live=True
 )