Update app.py

app.py

@@ -16,7 +16,7 @@ from datasets import load_dataset
 dataset = load_dataset("detection-datasets/coco", split="train", stream=True)
 
 # Reduce dataset to 20 rows, i.e., get sample
-samples = list(dataset.take(20))
+samples = dataset[:20]
 
 #Convert to dataframe
 df = pd.DataFrame(samples)