Spaces:

Dricz
/

Weapon_Detection_YOLOv8-4

Running

Dricz commited on Mar 21

Commit

381f849

•

1 Parent(s): 4d2ba31

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -134,7 +134,7 @@ def process_video(video_path):
     while cap.isOpened():
         ret, frame = cap.read()
         if not ret:
-            breakformat
         pil_img = Image.fromarray(frame[..., ::-1])
         result = model.predict(source=pil_img)

     while cap.isOpened():
         ret, frame = cap.read()
         if not ret:
+            break
         pil_img = Image.fromarray(frame[..., ::-1])
         result = model.predict(source=pil_img)