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 |
|