Update app.py

app.py

@@ -16,13 +16,6 @@ with st.sidebar:
     if video == 'YouTube':
         link_youtube = st.text_input('Masukan Link Youtube', '')
 
-# Process video
-if st.button('Start'):
-    if data_video == 'YouTube' and youtube_link:
-        process_video(youtube_link, model)
-    elif data_video == 'Webcam':
-        process_video('Webcam', model)
-
 # Load Model YOLOv9 
 model = YOLO('yolov9c.pt')
 
@@ -43,7 +36,7 @@ def process_video(data, model):
     if data == 'Webcam':
         camera = cv2.VideoCapture(0)  # kode webcam
     else:  # YouTube link
-        camera = CamGear(data=data, stream_mode=True, logging=True).start()
+        camera = CamGear(source=data, stream_mode=True, logging=True).start()
 
     FRAME_WINDOW = st.empty()
 
@@ -62,4 +55,11 @@ def process_video(data, model):
         FRAME_WINDOW.image(frame)
 
     if data == 'Webcam':
-        camera.release()
+        camera.release()
+
+# Process video
+if st.button('Start'):
+    if video == 'YouTube' and link_youtube:
+        process_video(link_youtube, model)
+    elif video == 'Webcam':
+        process_video('Webcam', model)