Update app.py

app.py

@@ -5,7 +5,7 @@ from ultralytics import YOLO
 
 
 def load_yolo_model():
-    return YOLO()
+    return YOLO('yolov8n-seg.pt')
 
 
 def process_video(model, cap):
@@ -39,12 +39,15 @@ def segment_webcam():
     process_video(model, cap)
 
 
-def wrapper_for_segment_video(uploaded_video):
-    segment_video(uploaded_video)
+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()
 
 
 iface = gr.Interface(
-    fn=None,
+    fn=process_inputs,
     inputs=[gr.inputs.Checkbox(label="Segment Video"),
             gr.inputs.Checkbox(label="Segment Webcam"),
             gr.inputs.File(label="Upload Video")],
@@ -52,17 +55,4 @@ iface = gr.Interface(
     live=True
 )
 
-
-def update_fn(segment_video_checkbox, segment_webcam_checkbox, uploaded_video):
-    if segment_video_checkbox and uploaded_video:
-        iface.fn = wrapper_for_segment_video
-        iface.args = [uploaded_video]
-    elif segment_webcam_checkbox:
-        iface.fn = segment_webcam
-        iface.args = []
-    else:
-        iface.fn = None
-
-
-iface.update(update_fn)
 iface.launch()