backup
Browse files
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",
|
|
|
|
|
|
|
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
|