AmandaPanda commited on
Commit
2f5cb18
·
verified ·
1 Parent(s): 33e5ff5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ import torch
13
  from datasets import load_dataset
14
 
15
  #Load dataset (detection-datasets/coco)
16
- dataset = load_dataset("henryscheible/coco_val2014_tiny", split="train")
17
 
18
  # Reduce dataset to 20 rows, i.e., get sample
19
  samples = dataset.select(range(20))
 
13
  from datasets import load_dataset
14
 
15
  #Load dataset (detection-datasets/coco)
16
+ dataset = load_dataset("henryscheible/coco_val2014_tiny")
17
 
18
  # Reduce dataset to 20 rows, i.e., get sample
19
  samples = dataset.select(range(20))