ktrapeznikov commited on
Commit
fe1bf49
·
verified ·
1 Parent(s): c74c9ba

Update script.py

Browse files
Files changed (1) hide show
  1. script.py +2 -2
script.py CHANGED
@@ -20,8 +20,8 @@ from preprocess import preprocess
20
 
21
 
22
  # load the dataset. dataset will be automatically downloaded to /tmp/data during evaluation
23
- SAFE_DATASET = os.environ.get("SAFE_DATASET","/tmp/data")
24
- dataset_remote = load_dataset(SAFE_DATASET,split = "test",streaming = True)
25
 
26
 
27
  # load your model
 
20
 
21
 
22
  # load the dataset. dataset will be automatically downloaded to /tmp/data during evaluation
23
+ DATASET_PATH = "/tmp/data"
24
+ dataset_remote = load_dataset(DATASET_PATH,split = "test",streaming = True)
25
 
26
 
27
  # load your model