Update app.py

app.py

@@ -40,7 +40,12 @@ SUBSAMPLE = 2
 def stream_object_detection(video, conf_threshold):
     cap = cv2.VideoCapture(video)
 
+    iterating = True
+    batch = []
+
+
     while iterating:
+        iterating, frame = cap.read()
         frame = cv2.resize( frame, (0,0), fx=0.5, fy=0.5)
         frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
         if n_frames % SUBSAMPLE == 0:
@@ -72,7 +77,6 @@ def stream_object_detection(video, conf_threshold):
             end = time.time()
             print("time taken for processing boxes", end - start)
 
-        iterating, frame = cap.read()
         n_frames += 1