Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -17,13 +17,15 @@ with zipfile.ZipFile("images/stable_diffusion.zip","r") as zip_ref:
|
|
17 |
def open_stable_ims(profession):
|
18 |
if len(profession) != 0:
|
19 |
dirname = 'images/stable_diffusion/'+ profession+'/'
|
20 |
-
images =
|
|
|
21 |
return images[:16]
|
22 |
|
23 |
def open_fair_ims(profession):
|
24 |
if len(profession) != 0:
|
25 |
dirname = 'images/fair_diffusion/' + profession+'/'
|
26 |
-
images =
|
|
|
27 |
return images[:16]
|
28 |
|
29 |
|
|
|
17 |
def open_stable_ims(profession):
|
18 |
if len(profession) != 0:
|
19 |
dirname = 'images/stable_diffusion/'+ profession+'/'
|
20 |
+
images = [Image.open(os.path.join(dirname+im)).convert("RGB") for im in os.listdir(dirname)]
|
21 |
+
random.shuffle(images)
|
22 |
return images[:16]
|
23 |
|
24 |
def open_fair_ims(profession):
|
25 |
if len(profession) != 0:
|
26 |
dirname = 'images/fair_diffusion/' + profession+'/'
|
27 |
+
images = [Image.open(os.path.join(dirname+im)).convert("RGB") for im in os.listdir(dirname)]
|
28 |
+
random.shuffle(images)
|
29 |
return images[:16]
|
30 |
|
31 |
|