vumichien commited on
Commit
f6f49c6
1 Parent(s): 2ff829b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -104,7 +104,7 @@ def preprocess_video(input_video_path):
104
 
105
  def predict(process_video):
106
  num_frames = int(cv2.VideoCapture(process_video).get(cv2.CAP_PROP_FRAME_COUNT))
107
- data_dir = tempfile.mkdtemp()
108
  tsv_cont = ["/\n", f"test-0\t{process_video}\t{None}\t{num_frames}\t{int(16_000*num_frames/25)}\n"]
109
  label_cont = ["DUMMY\n"]
110
  with open(f"{data_dir}/test.tsv", "w") as fo:
 
104
 
105
  def predict(process_video):
106
  num_frames = int(cv2.VideoCapture(process_video).get(cv2.CAP_PROP_FRAME_COUNT))
107
+
108
  tsv_cont = ["/\n", f"test-0\t{process_video}\t{None}\t{num_frames}\t{int(16_000*num_frames/25)}\n"]
109
  label_cont = ["DUMMY\n"]
110
  with open(f"{data_dir}/test.tsv", "w") as fo: