taesiri commited on
Commit
d03f6fb
·
1 Parent(s): a47b2ba
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -3,10 +3,17 @@ from datasets import load_dataset
3
  from PIL import Image
4
  import io
5
  import time
 
 
 
 
6
 
7
  # Load and prepare the dataset
8
  dataset = load_dataset(
9
- "taesiri/PhotoshopRequest-DailyDump", split="train", streaming=True
 
 
 
10
  )
11
 
12
  BUFFER_SIZE = 1
 
3
  from PIL import Image
4
  import io
5
  import time
6
+ import os
7
+
8
+
9
+ access_token = os.environ.get("HUGGINGFACE_TOKEN")
10
 
11
  # Load and prepare the dataset
12
  dataset = load_dataset(
13
+ "taesiri/PhotoshopRequest-DailyDump",
14
+ split="train",
15
+ streaming=True,
16
+ token=access_token,
17
  )
18
 
19
  BUFFER_SIZE = 1