itayitay123 commited on
Commit
d2550a4
·
verified ·
1 Parent(s): e329b4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -14,6 +14,11 @@ from PIL import Image
14
  # Load dataset from Hugging Face
15
  dataset = load_dataset("lirus18/deepfashion", split="train")
16
 
 
 
 
 
 
17
 
18
 
19
  from transformers import CLIPProcessor, CLIPModel
 
14
  # Load dataset from Hugging Face
15
  dataset = load_dataset("lirus18/deepfashion", split="train")
16
 
17
+ from PIL import Image
18
+
19
+ # Just to verify image loaded (not for UI)
20
+ image = dataset[0]['image']
21
+ image.save("sample_image.jpg") # saves one image to disk
22
 
23
 
24
  from transformers import CLIPProcessor, CLIPModel