Update app.py

app.py

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