Spaces:
Sleeping
Sleeping
Load different dataset (detection-datasets/coco
Browse files
app.py
CHANGED
|
@@ -10,7 +10,10 @@ from PIL import Image
|
|
| 10 |
import torch
|
| 11 |
|
| 12 |
# Get merve/coco dataset
|
| 13 |
-
|
|
|
|
|
|
|
|
|
|
| 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))
|