giovanni-bonetta commited on
Commit
ff62d04
·
1 Parent(s): 4621d8f
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -3,7 +3,7 @@ from datasets import load_dataset
3
  import tempfile
4
 
5
  # Load the dataset in streaming mode
6
- dataset = load_dataset("giobin/MAIA_2400")
7
 
8
  # Get an iterator over the dataset (assuming it's not too large)
9
  dataset_iter = iter(dataset["train"]) # Adjust split name if needed
@@ -12,5 +12,10 @@ dataset_iter = iter(dataset["train"]) # Adjust split name if needed
12
  video_example = next(dataset_iter)
13
  video_reader = video_example["video"] # Decord VideoReader object
14
 
 
 
 
 
 
15
  # Display the video
16
- st.video(video_reader[:].asnumpy().tobytes())
 
3
  import tempfile
4
 
5
  # Load the dataset in streaming mode
6
+ dataset = load_dataset("giobin/MAIA_2400", streaming=True)
7
 
8
  # Get an iterator over the dataset (assuming it's not too large)
9
  dataset_iter = iter(dataset["train"]) # Adjust split name if needed
 
12
  video_example = next(dataset_iter)
13
  video_reader = video_example["video"] # Decord VideoReader object
14
 
15
+ # Save the video temporarily
16
+ with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as tmp_file:
17
+ tmp_file.write(video_reader[:].asnumpy().tobytes()) # Convert frames to bytes
18
+ video_path = tmp_file.name
19
+
20
  # Display the video
21
+ st.video(video_path)