Change dataset to coco_val2014_tiny

app.py

@@ -13,10 +13,10 @@ import torch
 from datasets import load_dataset
 
 #Load dataset (detection-datasets/coco)
-dataset = load_dataset("henryscheible/coco_val2014_tiny", split="train", stream=True)
+dataset = load_dataset("henryscheible/coco_val2014_tiny", split="train")
 
 # Reduce dataset to 20 rows, i.e., get sample
-samples = list(dataset.take(20))
+samples = dataset.select(range(20))
 
 #Convert to dataframe
 df = pd.DataFrame(samples)