Update app.py

app.py

@@ -22,16 +22,24 @@ def get_video_length(file_path):
     return length
 
 def read_video_opencv(file_path, indices):
-    cap = cv2.VideoCapture(file_path)
     frames = []
-    for i in indices:
-        cap.set(cv2.CAP_PROP_POS_FRAMES, i)
-        ret, frame = cap.read()
-        if ret:
-            frames.append(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
-    cap.release()
+    with ThreadPoolExecutor() as executor:
+        futures = [executor.submit(get_frame, file_path, i) for i in indices]
+        for future in futures:
+            frame = future.result()
+            if frame is not None:
+                frames.append(frame)
     return frames
 
+def get_frame(file_path, index):
+    cap = cv2.VideoCapture(file_path)
+    cap.set(cv2.CAP_PROP_POS_FRAMES, index)
+    ret, frame = cap.read()
+    cap.release()
+    if ret:
+        return cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+    return None
+
 def sample_uniform_frame_indices(clip_len, seg_len):
     if seg_len < clip_len:
         repeat_factor = np.ceil(clip_len / seg_len).astype(int)
@@ -112,4 +120,4 @@ iface = gr.Interface(
     live=False
 )
 
-iface.launch(share=True)
+iface.launch()