Update app.py

app.py

@@ -4,6 +4,7 @@ import numpy as np
 import torch
 import clip
 import tempfile
+import cv2
 
 from tqdm import tqdm
 from transformers import GPT2Tokenizer
@@ -71,6 +72,8 @@ def main():
         tfile = tempfile.NamedTemporaryFile(delete=False)
         tfile.write(uploaded_file.read())
         st.write(tfile, tfile.name)
+        cap = cv2.VideoCapture(tfile.name)
+        st.write(cap)
     else:
         st.write('no')