Update

app.py

@@ -24,31 +24,32 @@ def process_frame(frame):
     segments = results[0].plot()
     return segments, f'FPS: {int(1 // (end - start))}'
 
-def process_inputs(start_button, stop_button, mode_selection, frame, uploaded_video):
+def process_inputs(action, mode_selection, frame, uploaded_video):
     global process, model
-    if start_button and mode_selection:
-        process = True
-        model = load_yolo_model()
+    if action == "Start" and not process and mode_selection:
         if mode_selection == "Webcam" and frame is not None:
+            process = True
+            model = load_yolo_model()
             return process_frame(frame)
         elif mode_selection == "Video" and uploaded_video is not None:
+            process = True
+            model = load_yolo_model()
             return process_frame(uploaded_video)
-    elif stop_button:
+    elif action == "Stop" and process:
         process = False
 
-start_button = gr.inputs.Button(label="Start")
-stop_button = gr.inputs.Button(label="Stop")
+action = gr.inputs.Dropdown(choices=["Start", "Stop"], label="Action")
 mode_selection = gr.inputs.Radio(["Webcam", "Video"], label="Mode Selection")
 frame = gr.inputs.Video(shape=(720, 1280), label="Webcam Feed")
 uploaded_video = gr.inputs.Video(shape=(720, 1280), label="Upload Video")
 
 iface = gr.Interface(
     fn=process_inputs,
-    inputs=[start_button, stop_button, mode_selection, frame, uploaded_video],
+    inputs=[action, mode_selection, frame, uploaded_video],
     outputs=[gr.outputs.Image(), gr.outputs.Textbox()],
     live=True,
     title="YOLO Image Segmentation",
-    description="This application uses the YOLO model to perform image segmentation on a webcam feed or an uploaded video. Select 'Webcam' or 'Video', upload a video (if applicable), and click 'Start' to begin. Click 'Stop' to end the process.",
+    description="This application uses the YOLO model to perform image segmentation on a webcam feed or an uploaded video. Select 'Webcam' or 'Video', upload a video (if applicable), and select 'Start' to begin. Select 'Stop' to end the process.",
     theme="huggingface"
 )