Spaces:

objectDetection
/

23A464A

Build error

apailang commited on Dec 27, 2023

Commit

b9015d3

1 Parent(s): d914368

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -83,7 +83,7 @@ def detect_video(video):
         if not ret:
             break
     for i in tqdm(range(nb_frames)):
-        ret, image_np = video_reader.read()
         input_tensor = tf.convert_to_tensor(np.expand_dims(image_np, 0), dtype=tf.uint8)
         results = detection_model(input_tensor)
         image_np_with_detections = viz_utils.visualize_boxes_and_labels_on_image_array(

         if not ret:
             break
     for i in tqdm(range(nb_frames)):
+        ret, image_np = cap.read()
         input_tensor = tf.convert_to_tensor(np.expand_dims(image_np, 0), dtype=tf.uint8)
         results = detection_model(input_tensor)
         image_np_with_detections = viz_utils.visualize_boxes_and_labels_on_image_array(