AmandaPanda commited on
Commit
fcc1af0
·
verified ·
1 Parent(s): d3f61c4

Load different dataset (detection-datasets/coco

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -10,7 +10,10 @@ from PIL import Image
10
  import torch
11
 
12
  # Get merve/coco dataset
13
- dataset = load_dataset("merve/coco", split='train', stream=True)
 
 
 
14
 
15
  # Reduce dataset to 20 rows, i.e., get sample
16
  samples = list(dataset_stream.take(20))
 
10
  import torch
11
 
12
  # Get merve/coco dataset
13
+ from datasets import load_dataset
14
+
15
+ #Load dataset (detection-datasets/coco)
16
+ dataset = load_dataset("detection-datasets/coco")
17
 
18
  # Reduce dataset to 20 rows, i.e., get sample
19
  samples = list(dataset_stream.take(20))